文章目录
- Question
- Ideas
- Code
Question
小明维护着一个程序员论坛。现在他收集了一份”点赞”日志,日志共有 N 行。
其中每一行的格式是:
ts id
表示在 ts 时刻编号 id 的帖子收到一个”赞”。
现在小明想统计有哪些帖子曾经是”热帖”。
如果一个帖子曾在任意一个长度为 D 的时间段内收到不少于 K 个赞,小明就认为这个帖子曾是”热帖”。
具体来说,如果存在某个时刻 T 满足该帖在 [T,T+D) 这段时间内(注意是左闭右开区间)收到不少于 K 个赞,该帖就曾是”热帖”。
给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
输入格式
第一行包含三个整数 N,D,K。
以下 N 行每行一条日志,包含两个整数 ts 和 id。
输出格式
按从小到大的顺序输出热帖 id。
每个 id 占一行。
数据范围
1≤K≤N≤105,
0≤ts,id≤105,
1≤D≤10000
输入样例:
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3
输出样例:
1
3
Ideas
遍历时间段,统计每个id的点赞数,如果时间段长度超了,那么需要移动时间段左指针
Code
n,d,k = list(map(int,input().strip().split()))
N = 100010
ste = [0 for i in range(N)] # 存储的是每个id的状态
nums = [0 for i in range(N)] # 存储的是每个id 获赞数
lis = [] # 存储的是输入
for i in range(n):
lis.append(list(map(int,input().strip().split())))
lis = sorted(lis,lambda x:(x[0],x[1]))
j = 0 # 时间段左指针
for i in range(n):
nums[lis[i][1]] += 1
while lis[i][0] - lis[j][0] >= d:
nums[lis[j][1]] -= 1
j += 1
if nums[lis[i][1]] >= k: # 这里不能直接输出的原因 [按从小到大的顺序输出热帖 id]
ste[lis[i][1]] = 1
for i in range(N):
if ste[i]:
print(i)