日志统计&&递增三元组

题目

小明维护着一个程序员论坛。现在他收集了一份"点赞"日志，日志共有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. 1≤i,j,k≤N;

2. 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)