CCF-CSP 202209-3 防疫大数据

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

//存到风险名单里的三个条件：在近七日里曾经出现在漫游数据中，在七日内有到访某个地区的记录,近七日到访的地区在到访的那一日持续处于风险状态，并且从到访日到名单生成日持续处于风险状态
//站在生成名单当天的角度来讲，不考虑某个地区已经解除风险，且接收到了某个用户在之前那个风险段去过这个地区的情况，因为要求到访到生成日全处于风险状态，已经断掉了。
struct message {
	int day;
	int user;
	int region;
	message(int day, int user, int region): day(day), user(user), region(region) {}
};

int main() {
	int n;//总数据天数
	cin >> n;
	unordered_map<int, pair<int, int>>regions; //记录地区号，风险开始日到风险结束日
	vector<message>messages[1010];//按天数储存所有有判断价值的漫游消息
	vector<int>ans[1010];//按天数储存每日风险用户名单（记得去重），只要求存当天有风险的用户编号，不要求地区
	for (int d = 0; d < n; d++) {
		int r, m;
		cin >> r >> m; //分别表示收到风险地区信息的数量和收到的漫游数据的数量
		while (r--) { //收到风险地区的列表
			int reg;
			cin >> reg;
			if (regions.count(reg) == 0
			        || d - regions[reg].second > 1)//注意是大于一不是大于零，前一天是有风险的最后一天，后一天是有风险的第一天仍然算连续
				regions[reg].first = d; //更改开始日期
			regions[reg].second = d + 6;
		}
		while (m--) {//收到的漫游消息列表
			int d1, u1, r1; //某一组漫游数据,分别表示到访日，用户编号，地区编号
			cin >> d1 >> u1 >> r1;
			if (d - d1 > 6 || regions.count(r1) == 0 || d > regions[r1].second)//过期消息或所到地区无风险
				continue;//这些消息没必要存下来，今天用不到，之后的判断也用不到
			messages[d].push_back(message(d1, u1, r1));
		}
		for (int i = max(0, d - 6); i <= d; i++) {//在近七日里曾经出现在漫游数据中
			for (int j = 0; j < messages[i].size(); j++) { //遍历到访天数为i的用户的信息列表
				int r2 = messages[i][j].region;
				int d2 = messages[i][j].day;
				int u2 = messages[i][j].user;
//近七日到访的地区在到访的那一日持续处于风险状态，并且从到访日到名单生成日持续处于风险状态
				if (regions[r2].first <= d2 && regions[r2].second >= d && regions.count(r2) && d - d2 <= 6)
					ans[d].push_back(u2);
			}
		}
	}

	for (int i = 0; i < n; i++) {
		sort(ans[i].begin(), ans[i].end());
		ans[i].erase(unique(ans[i].begin(), ans[i].end()), ans[i].end());
		cout << i ;
		for (int j = 0; j < ans[i].size(); j++)
			cout << " " << ans[i][j];
		cout << endl;
	}
	return 0;

}