CSP 202203-3计算资源调度器

问题分析：

根据题目描述，可以确定一个可用区可以包含很多节点，每个节点只在一个可用区。而且由于可用区和节点数都小于等于1000，比较小。所以可以按照下面的形式定义可用区以及节点。因为任务编号比较大，为了方便起见，采用map来存储每个可用区及节点中的任务信息。

map<int,int> region[1005];//可用区中的任务信息。
struct Node//节点
{
	int r; // 所属可用区
	int cnt;
	map<int,int> vis;//该节点下的任务
	int index;
};
Node node[1005];

对于三个需求，可以对每个需求寻找合法的节点，最后再寻找满足三个需求条件的节点。

void insert1(int na) //需求1
{
	if(na == 0) //所有节点都满足
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(node[i].r == na)// 节点是否在要求的可用区内
				vis[i]++;
	}
}
void insert2(int pa) // 需求2
{
	if(pa == 0)//所有节点都满足
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(region[node[i].r][pa] == 1)
			//目标任务所在可用区下的所有节点都满足。
				vis[i]++;
	}
}
void insert3(int paa) //需求3
{
	if(paa == 0)//所有节点都满足
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(node[i].vis.find(paa) == node[i].vis.end())
			//目标任务所在可用区下的所有节点都不满足
				vis[i]++;
	}
}

所以统计完之后，可以判断满足条件的节点信息。

	insert1(na);
	insert2(pa);
	insert3(paa);
	vector<Node> tempans;
	for(int i=1;i<=n;i++)
	{
		if(vis[i] == 3) // 满足三个条件可以分配
			tempans.push_back(node[i]);
	}

至此过滤操作已经完成，排序操作相对比较简单。因为我们只需要找一个最优的节点，所以按照题目排序要求将合法节点序列遍历一遍即可。

int maxNode(vector<Node> tnode)
{
	int ans = tnode[0].index;
	for(int i=1;i<tnode.size();i++)
	{
		if(tnode[i].cnt < node[ans].cnt)
			ans = tnode[i].index;
		else if(tnode[i].cnt == node[ans].cnt && tnode[i].index<node[ans].index)
			ans = tnode[i].index;
	}
	return ans;
}

选定节点之后，对答案进行输出即可。

void insertNode(int a, int index)
{
	node[index].vis[a] = 1;
	node[index].cnt++;
	region[node[index].r][a] = 1;
	cout<< index <<' ';
}

如果没有满足条件的节点，并且计算任务反亲和性要求是尽量满足的，那么我们可以把vis数组清空，然后重新计算一下即可。

		insert1(na);
		insert2(pa);
		for(int i=1;i<=n;i++)
		{
			if(vis[i] == 2) // 可以分配
				tempans.push_back(node[i]);
		}

完整代码

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long 
#define ld long double
#define gcd(a,b) __gcd(a,b)
using namespace std;
const int INF = INT_MAX; 
int n,m,g;
map<int,int> region[1005];
struct Node//节点
{
	int r; // 所属可用区
	int cnt;
	map<int,int> vis;
	int index;
};
Node node[1005];
int vis[1005] = {0};
void insert1(int na)
{
	if(na == 0)
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(node[i].r == na)
				vis[i]++;
	}
}
void insert2(int pa)
{
	if(pa == 0)
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(region[node[i].r][pa] == 1)
				vis[i]++;
	}
}
void insert3(int paa)
{
	if(paa == 0)
	{
		for(int i=1;i<=n;i++)
			vis[i]++;
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(node[i].vis.find(paa) == node[i].vis.end())
				vis[i]++;
	}
}
int maxNode(vector<Node> tnode)
{
	int ans = tnode[0].index;
	for(int i=1;i<tnode.size();i++)
	{
		if(tnode[i].cnt < node[ans].cnt)
			ans = tnode[i].index;
		else if(tnode[i].cnt == node[ans].cnt && tnode[i].index<node[ans].index)
			ans = tnode[i].index;
	}
	return ans;
}
void insertNode(int a, int index)
{
	node[index].vis[a] = 1;
	node[index].cnt++;
	region[node[index].r][a] = 1;
	cout<< index <<' ';
}
void insert(int a,int na,int pa,int paa,int paar)
{
	memset(vis, 0, sizeof(vis));
	insert1(na);
	insert2(pa);
	insert3(paa);
	vector<Node> tempans;
	for(int i=1;i<=n;i++)
	{
		if(vis[i] == 3) // 可以分配
			tempans.push_back(node[i]);
	}
	if(tempans.size()!=0)
	{
		int index = maxNode(tempans);
		insertNode(a, index);
		return;
	}
	if(paa != 0 && paar==0)
	{
		memset(vis,0, sizeof(vis));
		insert1(na);
		insert2(pa);
		for(int i=1;i<=n;i++)
		{
			if(vis[i] == 2) // 可以分配
				tempans.push_back(node[i]);
		}
	}
	if(tempans.size()!=0)
	{
		int index = maxNode(tempans);
		insertNode(a, index);
		return;
	}
	cout<<0<<' ';
}
void solve()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		int a;
		cin>>a;
		node[i].r = a;
		node[i].index = i;
	}
	cin>>g;
	for(int i=0;i<g;i++)
	{
		int f,a,na,pa,paa,paar;
		cin>>f>>a>>na>>pa>>paa>>paar;
		while(f--)
		{
			insert(a, na, pa, paa, paar);
		}
		cout<<endl;
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	solve();
	return 0;
}