[每日一题]投资结果

Posted1 by 呼延十 on November 28, 2018 Hot:

来源:

lintcode-投资结果

描述

给定一个列表funds表示投资人每次的投资额。现在有三个公司A, B, C,它们的初始资金分别为a,b,c。投资人每次投资时会对当前资金最少的公司进行投资(当有多个公司资金相同时,投资人会对编号最小的公司进行投资)。返回A, B, C三家公司最后的资金。

样例

  1. 给定funds=[1,2,1,3,1,1],a=1,b=2,c=1, 返回[4,5,4]
第一次投资时A和C的资金相同,选择对编号较小的A投资,此时a=2, b=2, c=1
第二次投资时C的资金最少,对C进行投资,此时a=2, b=2, c=3
第三次投资时A和B的资金相同,选择对编号较小的A投资,此时a=3, b=2, c=3
第四次投资时B的资金最少,对B进行投资,此时a=3, b=5, c=3
第五次投资时A和C的资金相同,选择对编号较小的A投资,此时a=4, b=5, c=3
第六次投资时C的资金最少,对C进行投资,此时a=4, b=5, c=4
  1. 给定funds=[2,1,1,1],a=1,b=2,c=2, 返回[4,3,3]
第一次投资时A的资金最少,对A进行投资,此时a=3, b=2, c=2
第二次投资时B和C的资金相同,选择对编号较小的B投资,此时a=3, b=3, c=2
第三次投资时C的资金最少,对C进行投资,此时`a=3, b=3, c=3
第四次投资时A, B和C的资金相同,选择对编号较小的`A`投资,此时a=4, b=3, c=3

解题思路

就按照最常见得思路来遍历吧,我没想到更好点的办法…

实现代码:

public int[] getAns(int[] funds, int a, int b, int c) {
  // Write your code here
  //将abc放入int数组,相当于给abc添加索引
  int[] arr = new int[]{a, b, c};
  for (int i = 0; i < funds.length; i++) {
    //给正确的位置+资金
    arr[judge(arr)] += funds[i];
  }
  return arr;
}

/**
 * 获取正确的位置 先比较a,b,再用小的值和c比较,因为只需要去最小.
 *
 * @param arr 当前的abc数组
 * @return 正确位置的索引
 */
public int judge(int[] arr) {
  //默认为a投资
  int i = 0;
  //如果a>b,给b投资
  if (arr[0] > arr[1]) {
    i = 1;
  }
  //上一步算出的结果如果大于c,给c投资
  if (arr[i] > arr[2]) {
    i = 2;
  }
  return i;
}

这道题的主要注意事项是,给a,b,c添加索引,防止真的去拿a,b,c去比较.然后再获取正确的索引方法中,也可以想办法继续优化.

完。




ChangeLog

2018-11-28 完成

以上皆为个人所思所得,如有错误欢迎评论区指正。

欢迎转载,烦请署名并保留原文链接。

联系邮箱:huyanshi2580@gmail.com

更多学习笔记见个人博客——>呼延十