【HackerRank】Bus Station

有n组好朋友在公交车站前排队。第i组有ai个人。还有一辆公交车在路线上行驶。公交车的容量大小为x，即它可以同时运载x个人。 当车站来车时(车总是空载过来)，一些组从会队头开始走向公交车。 当然，同一组的朋友不想分开，所以仅当公交车能容纳下整个组的时候，他们才会上车。另外，每个人不想失去自己的位置，即组的顺序不会改变。 问题时如何选择公交车的容量大小x使得它可以运走所有组的人，并且公交车每次从车站出发时没有空位？（在车里的总人数恰好达到x)?

输入格式

第一行只包含一个整数n。第二行包含n个空格分隔的整数a1,a2,…,an。

输出格式

按递增顺序输出所有可能的公交车的容量。

主要注意两点：

• 枚举total_people因子的时候，只用从1枚举到sqrt(total_people)，因为知道total_people的一个因子r，就可以通过total_people/r得到另外一个因子了。要特别处理的情况是r=total_people/r的情况，此时只留一个因子；
• 还要注意最后一趟上车要保证剩下的人全部上车（参见代码17行）。

代码如下：

 1 import java.io.*;
 2 import java.util.*;
 3 import java.math.*;
 4
 5
 6 public class Solution {
 7     private static boolean Check(int[] groups,int d){
 8         int curPeople = 0;
 9         for(int i = 0;i < groups.length;i++)
10         {
11             curPeople += groups[i];
12             if(curPeople > d)
13                 return false;
14             if(curPeople == d)
15                 curPeople = 0;
16         }
17         return curPeople == 0;
18     }
19     public static void main(String[] args) {
20         Scanner in = new Scanner(System.in);
21         int n = in.nextInt();
22         int total_people = 0;
23         int[] groups = new int[n];
24
25         for(int i = 0;i < n;i++){
26             groups[i]= in.nextInt();
27             total_people += groups[i];
28         }
29
30         ArrayList<Integer> answer = new ArrayList<Integer>();
31
32         for(int d = 1;d*d <= total_people;d++){
33             if(total_people % d == 0){
34                 if(Check(groups, d))
35                     answer.add(d);
36                 if(total_people/d != d && Check(groups, total_people/d))
37                     answer.add(total_people/d);
38             }
39         }
40         answer.sort(null);
41
42         StringBuffer sb = new StringBuffer();
43         for(Integer i:answer)
44             sb.append(i).append(" ");
45         System.out.println(sb.toString());
46
47       }
48 }

【HackerRank】Bus Station,布布扣,bubuko.com