4月18日，腾讯笔试第二题发广告

4月18日，腾讯笔试第二题发广告

 输入:
 n	m       //n 表示用户数量， m表示需要发广告数）
 t1			//n 组输入，ti表示 给第 i 个用户发广告的时间，在ti的倍数时也需要给用户 i 在发一份广告
 ...		//时间相撞时， 先输出 id 较小的用户
 tn			// 数据范围 0 <= n,m <= 10^6, 0 <= t1 ... tn <= 1000
   
 案例
 3	5
 3
 6
 9
   
 输出： （第三秒输出 1， 第六秒时输出 1，2，第 9 秒时输出1，3）
 1
 1
 2
 1
 3

• 当时一下没做起来，结束后整理一下思路写出来了， 不知道能不能A全部数据
• 思路：
  • 维护一个 list 集合，list 里面存放 int[] , int[0] 表示给每一个用户可能发广告的时间，int[1] 表示对应的用户id，也就是下标
  • 全部存好后，对 lists 进行排序， 先按时间升序排列， 后按用户 ID 升序排列
  • 遍历排好序后的 lists， 输出前 m 个 ID 即解。

import java.util.*;
public class Main {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
      //n : 用户数， m ： 代发广告数
        int n = sc.nextInt();
        int m = sc.nextInt();
		
        List<int[]> lists = new ArrayList<>();
      //第一层循环遍历用户
        for(int i = 1 ; i <= n ; i++){
            int t = sc.nextInt();
          //第二层循环遍历所有可能给该用户发广告的时间
          //全部存进 lists 集合中
            for(int j = t ; j <= t * m ; j += t){
                lists.add(new int[] {j,i});
            }
        }

      //先按时间升序排序， 后按用户 ID 升序排序
      //最后从小到大输出 m 个 lists 元素的 ID 
      //即可达到效果
        Collections.sort(lists, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                //按时间升序
                int flag = o1[0] - o2[0];

                //时间相同时， 按id升序
                if(flag == 0){
                    return o1[1] - o2[1];
                } else {
                    return flag;
                }
            }
        });

        for(int i = 0 ; i < m ; i++){
            System.out.println(lists.get(i)[1]);
        }

    }
}