CCF-CSP 202203-3 计算资源调度器

#include <bits/stdc++.h>
using namespace std;
//帮每个任务找一个合适的节点
//计算节点亲和性，可以记录每个区有什么节点，用一个无序map
//计算任务亲和性，可以记录每个应用的任务在哪些节点上运行，用一个无序map，前一项是应用编号，后一个是set，记录任务编号
//计算任务反亲和性，用亲和性的数据结构即可满足要求
//要排序，记录每个节点已经部署的任务数量即可
//题目要实现什么就定义什么样的数据结构，没必要冗余定义；判断单独写函数，不然写if else的判断真的很烦人
//unordered_map<int, int>field;//前一个是节点编号，后一个是对应区域编号
unordered_map<int, unordered_set<int>>apply; //前一个是应用编号，后一个是应用的任务部署的节点的编号
unordered_map<int, int>task; //前一个是节点编号，后一个是该节点运行的任务数量

int field[1001] = {0}; //前一个是节点编号，后一个是对应区域编号

bool checkpoint(int nai, int pointnum) { //检查某个节点是不是符合节点亲和性
	if (nai == 0 || field[pointnum] == nai)
		return true;
	return false;
}

bool checktask(int pai, int pointnum) { //检查某个节点是否满足任务亲和性，即这个节点所在的区域里面有没有指定应用的任务
	if (pai == 0)
		return true;
	int fieldnum = field[pointnum]; //检查的节点所在的区域编号，检查这个区域有没有指定应用的任务
	for (auto i : apply[pai]) { //i是指定应用的任务部署的节点编号
		if (fieldnum == field[i])
			return true;//如果有一个指定应用的任务所在节点与之同区域 返回true
	}
	return false;
}

bool checknotask(int paai, int pointnum) {//检查不能和指定应用的任务在同一个节点上运行，通过应用找节点就好了
	if (paai == 0)
		return true;
	for (auto i : apply[paai]) { //i是指定应用的任务部署的节点编号
		if (i == pointnum)
			return false;//如果有一个指定应用的任务所在节点与之同区域 返回false
	}
	return true;
}

int main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int pointnumber, fieldnumber;
	cin >> pointnumber >> fieldnumber;
	for (register int i = 1; i <= pointnumber; i++) { //i其实就是节点编号
		int temp;
		cin >> temp;
		//field.emplace(i, temp);
		field[i] = temp;
	}
	int group;
	cin >> group;
	for (register int g = 0; g < group; g++) {
		int fi, ai, nai, pai, paai, paari;
		cin >> fi >> ai >> nai >> pai >> paai >> paari;
		//任务数量 任务所属应用编号 节点亲和性要求 任务亲和性要求 任务反亲和性要求 必须还是尽量
		for (register int f = 0; f < fi; f++) {
			set<int>ans;//严格满足三个特性
			set<int>ans1;//满足前两个特性，如果非必要且ans为空也可以拿来当解集
			for (register int i = 1; i <= pointnumber; i++) { //遍历每个节点编号
				//int pointnum = i.first;
				//int field = i.second;
				if (checkpoint(nai, i) && checktask(pai, i)) {
					if (checknotask(paai, i))
						ans.insert(i);
					ans1.insert(i);
				}
			}
			if (ans.size() == 0 && paari == 0) //遍历完后发现没有节点同时满足三个要求,而且是尽量满足
				ans = ans1;

			//把ans按照任务量和编号排序
			if (ans.size() == 0) {
				cout << 0 << " ";
				continue;
			} else {
				vector<pair<int, int>>node; //前一个参数代表任务数量，后一个参数代表节点编号
				for (auto i : ans)
					node.push_back(make_pair(task[i], i));
				sort(node.begin(), node.end());
				int result = node[0].second; //当前任务最终储存的节点编号
				cout << result << " ";
				apply[ai].insert(result);
				task[result]++;
			}
		}
		cout << '\n';
	}
	return 0;
}