思路:
根据每天输入的风险区更新风险区列表map<int, set<int>> area,主键为日期,存储该日期下的风险区列表。同样的形式存储结果列表,以multiset存储漫游数据。遍历每天,更新风险区列表,将漫游数据存储。首先去除六天前的漫游数据,然后检测从漫游数据当天到今天该条漫游数据去的地点是否仍是风险区,注意风险区需连续,如示例第六天到第八天中间1不是风险区。
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int n, ri, mi, pij, d, u, a;
map<int, set<int> > area;
map<int, set<int> > p;
struct mdata {
int d, u, r;
void read() {
cin >> d >> u >> r;
}
bool operator<(const mdata &a)const {
return d < a.d;
}
};
multiset<mdata> routes_data;
void clear_expired_day(int today) {
auto limit = routes_data.lower_bound({today - 6});
routes_data.erase(routes_data.begin(), limit);
}
bool is_risk_area(int today, mdata m){
for(int i = m.d; i<= today; i++){
if(area[i].count(m.r) == 0)
return false;
}
return true;
}
int main() {
ios::sync_with_stdio(false);
cin >> n;
for(int i=0; i<n; i++) {
cin >> ri >> mi;
for(int j=0; j<ri; j++) {
cin >> pij;
area[i].insert(pij);
for(int k = 1; k<7; k++) {
area[i+k].insert(pij);
}
}
for(int j=0; j< mi; j++) {
mdata md;
md.read();
routes_data.insert(md);
}
clear_expired_day(i);
for(auto it : routes_data) {
if(is_risk_area(i, it)) {
p[i].insert(it.u);
}
}
}
for(int i=0; i<n; i++) {
cout << i << " ";
for(auto it : p[i]) {
cout << it << " ";
}
cout << endl;
}
return 0;
}
参考题解:CSP 202209题解:如此编码,何以包邮,防疫大数据,吉祥物投票,高维亚空间超频物质变压缩技术_吉祥物投票csp-CSDN博客