【CCF CSP历年考试带注释全题解 C++版】201912-4 区块链

#include <iostream>
#include <vector>
#include <map>
#include <unordered_map>
using namespace std;

vector<vector<int>> connection; //记录节点邻居
vector<vector<int>> chain; //记录各节点主链
map<int, unordered_map<int, vector<int>>> update_chain; //时间->（目标节点->用来比较更新的主链)
map<int, unordered_map<int, vector<int>>> add_block; //时间->（目标节点->添加的新块(可能在同一时间为同一节点添加多个块)）

bool superior(vector<int>& chain_a, vector<int>& chain_b) { //用于比较两条主链的优先级，优先级高在下文称为“大”
	int s1 = chain_a.size(), s2 = chain_b.size();
	return (s1 > s2 || (s1 == s2 && chain_a[s1 - 1] < chain_b[s2 - 1]));
}

void update(vector<int>& new_chain, int time, int target) { //用于减少更新次数，
	if (superior(new_chain, chain[target])) { //只有新链比旧链大，
		if (update_chain.count(time) && update_chain[time].count(target)) { //且比同一时间的新链大
			if (superior(new_chain, update_chain[time][target])) {
				update_chain[time][target] = new_chain;
			}
		}
		else update_chain[time][target] = new_chain; //或这一时刻的新链不存在时才加入队列
	}
}

int main() {
	int n, m;
	cin >> n >> m; //输入n，m
	connection = vector<vector<int>>(n + 1); //记录邻居关系
	chain = vector<vector<int>>(n + 1, vector<int>{0}); //记录各节点主链
	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		connection[u].push_back(v);
		connection[v].push_back(u);
	}
	int t, k;
	cin >> t >> k; //输入t，k
	
	for (int i = 0; i < k; i++) {
		int a = -1, b = -1, c = -1;
		cin >> a >> b; //读取a，b
		if (cin.peek() != '\n') cin >> c; //若下一个字符不是换行符,读取c
		if (c != -1) { //若有c输入，则将添加块的操作添加进映射
			add_block[b][a].push_back(c);
		}
		else {
			while (!update_chain.empty() || !add_block.empty()) { //当更新队列和添加队列都不为空时
				int first_update = update_chain.empty() ? INT_MAX : update_chain.begin()->first; //排在最前面的更新链时间，若为空集则取最大值
				int first_add = add_block.empty() ? INT_MAX : add_block.begin()->first; //排在最前面的添加块时间，若为空集则取最大值
				if (first_update > b && first_add > b) break; //都为空集则跳出循环
				else if (first_update <= first_add) { //取等号是为了两者时间相同时优先处理update
					for (auto i = update_chain[first_update].begin(); i != update_chain[first_update].end(); i++) { //某一时间的所有更新链操作
						if (superior(i->second, chain[i->first])) { //如果新链比旧链大，
							chain[i->first] = i->second; //替换旧链，
							for (int neighbour : connection[i->first]) { //并发送新链给所有邻居，
								update(chain[i->first], first_update + t, neighbour); //这里update会筛选更大的链发给邻居
							}
						}
					}
					update_chain.erase(update_chain.begin()); //此时更新链操作全部完成，直接删除第一个操作
				}
				else {
					for (auto i = add_block[first_add].begin(); i != add_block[first_add].end(); i++) { //某一时间的所有添加块操作
						chain[i->first].insert(chain[i->first].end(), i->second.begin(), i->second.end()); //在旧链末尾添加块
						for (int neighbour : connection[i->first]) { //发送新链给所有邻居
							update(chain[i->first], first_add + t, neighbour);
						}
					}
					add_block.erase(add_block.begin()); //此时添加块操作全部完成，直接删除第一个操作
				}
			}
			cout << chain[a].size() << " "; //输出结果
			for (int block : chain[a]) {
				cout << block << " ";
			}
			cout << endl;
		}
	}
}