题目
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有N 行。其中每一行的格式是:
ts id
表示在ts 时刻编号id 的帖子收到一个"赞"。
现在小明想统计有哪些帖子曾经是"热帖"。如果一个帖子曾在任意一个长度为D 的时间段内收到不少于 K 个赞,小明就认为这个帖子曾是"热帖"。
具体来说,如果存在某个时刻 T 满足该帖在[T,T+D) 这段时间内(注意是左闭右开区间)收到不少于K 个赞,该帖就曾是"热帖"。
给定日志,请你帮助小明统计出所有曾是"热帖"的帖子编号。
思路
将每个日志用字典存储起来,key为id,value为对该id点赞的时间列表。对每个id点赞的时间list进行判断,每个点赞记录向后移动k个,即找到以当前点赞时间往后推k-1个点赞,如i+k-1个点赞时间与i个点赞时间的差距为D,则表示该帖子为热帖。按此思路判断所有帖子。
代码
import os
import sys
# 请在此输入您的代码
N, D, K = map(int, input().split())
logs = {}
for i in range(N):
ts, idd = map(int, input().split())
if str(idd) not in logs.keys():
logs[str(idd)] = [ts]
else:
logs[str(idd)].append(ts)
hot_id = []
for key, ts in logs.items():
temp_ts = sorted(ts)
for i in range(len(temp_ts)-K+1):
if temp_ts[i+K-1] < temp_ts[i] + D:
hot_id.append(int(key)) #这里添加的是str 如果不修改为int型 则会有5个用例报错 应该是输出的格式问题 但其实dict中的key可以是int型
break
for i in sorted(hot_id):
print(i)
题目
给定三个整数数组
A=[A1,A2,⋯AN],
B=[B1,B2,⋯BN],
C=[C1,C2,⋯CN],
请你统计有多少个三元组(i,j,k) 满足:
1≤i,j,k≤N;
Ai<Bj<Ck。
思路
对B数组中的每个数进行判断,找到A中比该数小的个数 * C中比该数大的个数。
代码
import os
import sys
# 请在此输入您的代码
N = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
C = list(map(int, input().split()))
A.sort()
B.sort()
C.sort()
i, k = 0, 0
count = 0
for j in range(N):
while i < N and A[i] < B[j]:
i += 1
while k < N and C[k] <= B[j]:
k += 1
count += i * (N-k)
print(count)