第九屆藍橋杯Java B——日誌統計
小明維護著一個程式設計師論壇。現在他收集了一份"點贊"日誌,日誌共有N行。其中每一行的格式是:
ts id
表示在ts時刻編號id的帖子收到一個"贊"。
現在小明想統計有哪些帖子曾經是"熱帖"。如果一個帖子曾在任意一個長度為D的時間段內收到不少於K個贊,小明就認為這個帖子曾是"熱帖"。
具體來說,如果存在某個時刻T滿足該帖在[T, T+D)這段時間內(注意是左閉右開區間)收到不少於K個贊,該帖就曾是"熱帖"。
給定日誌,請你幫助小明統計出所有曾是"熱帖"的帖子編號。
【輸入格式】
第一行包含三個整數N、D和K。
以下N行每行一條日誌,包含兩個整數ts和id。
對於50%的資料,1 <= K <= N <= 1000
對於100%的資料,1 <= K <= N <= 100000 0 <= ts <= 100000 0 <= id <= 100000
【輸出格式】
按從小到大的順序輸出熱帖id。每個id一行。
【輸入樣例】
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3
【輸出樣例】
1
3
資源約定:
峰值記憶體消耗(含虛擬機器) < 256M
CPU消耗< 1000ms
import java.io.BufferedInputStream; import java.util.ArrayList; import java.util.Collections; import java.util.HashSet; import java.util.Map; import java.util.Scanner; import java.util.Set; import java.util.TreeMap; public class Main { public static void main(String[] args) { Scanner cin = new Scanner(new BufferedInputStream(System.in)); int n = cin.nextInt(); int d = cin.nextInt(); int k = cin.nextInt(); Map<Integer, ArrayList<Integer>> map = new TreeMap<Integer, ArrayList<Integer>>(); for (int i = 0; i < n; i++) { int ts = cin.nextInt(); int id = cin.nextInt(); ArrayList<Integer> list = new ArrayList<Integer>(); if (!map.containsKey(id)) { list.add(ts); map.put(id, list); } else { list = map.get(id); list.add(ts); map.put(id, list); } } Set<Integer> res = new HashSet<Integer>(); for (int id : map.keySet()) { Collections.sort(map.get(id)); int sum = 1; int start = map.get(id).get(0); for (int i = 1; i < map.get(id).size(); i++) { if (map.get(id).get(i) - start < d) { sum++; } else if (map.get(id).get(i) - start > d) { start = map.get(id).get(i); } if (sum >= k) { res.add(id); break; } } } for (int i : res) System.out.println(i); } }