方法:哈希表+滑动窗口[left,right]
y总做法:
//给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
/**
* 算法1:滑动窗口 + 哈希表
时间复杂度:O(NlogN)
空间复杂度:O(N)
首先按照时间戳对日志进行排序,然后使用一个哈希表记录每个帖子在当前时间点之前的总点赞数,
使用一个滑动窗口来遍历日志。
对于每一个时间点 ts,我们将其加入滑动窗口中,同时更新哈希表中对应帖子的点赞数。
对于滑动窗口中的每一个帖子,
如果其在滑动窗口内获得了不少于 K 个赞,则认为该帖子是热帖,加入结果集。
由于滑动窗口和哈希表都是线性复杂度操作,
因此总时间复杂度为 O(NlogN)。
*/
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 100010;
int cnt[N];
bool st[N]; // 记录每个帖子是否是热帖
struct Post {
int id, ts;
};
bool cmp(Post a, Post b) {
return a.ts < b.ts;
}
int main() {
int n, d, k;//n行数据,[t,t+d]这区间内,点赞数不小于k
cin >> n >> d >> k;
///数据录入
vector<Post> posts(n);
for (int i = 0; i < n; i++) {
cin >> posts[i].ts >> posts[i].id;//时间戳 id
}
sort(posts.begin(), posts.end(), cmp);//按时间戳排序
//debug
/*cout<<'t'<<' '<<"id"<<endl;
for (int i = 0; i < n; i++) {
cout<< posts[i].ts <<' '<< posts[i].id<<endl;//时间戳 id
}
cout<<endl;*/
for (int i = 0 ,j = 0; i < n; i++) {
int id = posts[i].id;
cnt[id] ++ ;//[i,i+d]区间内,id出现的次数
//logs[i].x - logs[j].x >= d
//logs[i].x >= d + logs[j].x
//d + logs[j].x <= logs[i].x
//logs[j].x <= logs[i].x -d
while (j < n && posts[j].ts <= posts[i].ts - d) {//j枚举id
//cout<<posts[j].id<<"出局"<<endl;
cnt[ posts[j].id ] -- ;//可以为负数
j++;
}
if (cnt[id] >= k) st[id] = true;
}
for (int i = 0; i < 100000; i++) {
if (st[i])
printf("%d\n", i);
}
return 0;
}
chatgpt做法
#include <iostream>
#include <unordered_map>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
int main()
{
int n, d, k;
cin >> n >> d >> k;
vector<pair<int, int>> logs(n);
for (int i = 0; i < n; i ++ ) cin >> logs[i].first >> logs[i].second;
//1.将所有的点赞记录按照时间戳进行排序(升序排列)。
sort(logs.begin(), logs.end());
//2.维护一个哈希表,记录当前时间窗口内每个帖子被点赞的次数。
unordered_map<int, int> cnt;//哈希表负责增删操作
set<int> hot;
int left = 0;
for (int i = 0; i < n; i ++ )//遍历所有点赞记录,同时对哈希表进行增删操作 i是右指针
{
cnt[logs[i].second] ++ ;//右指针,从右到左遍历,每个时间戳对应的id都会在哈希中记录一次
//在[ts,ts+D]窗口内进行以下操作:
//3.使用双指针维护一个长度为 D 的时间窗口,遍历所有的点赞记录。
while (left <= i && logs[i].first - logs[left].first >= d)//维护[left,i]这个窗口,直到logs[i].first - logs[left].first < d时,窗口失效
{
//说明:logs[i].first - logs[left].first >= d
cnt[logs[left].second] -- ;
left ++ ;
}
//4.如果某个帖子收到的赞数不小于K,则将其加入结果集中。
if (cnt[logs[i].second] >= k) hot.insert(logs[i].second);
}
for (auto x : hot) cout << x << endl;
return 0;
}