Codeforces Round #237 (Div. 2) C. Restore Graph(图论+贪心)

题目链接
Codeforces Round #237 (Div. 2) C. Restore Graph(图论+贪心)_第1张图片
Codeforces Round #237 (Div. 2) C. Restore Graph(图论+贪心)_第2张图片
思路:这是一道披着图论外衣的简单贪心,要想有解首先d【i】=0必须只有一个,其次排序后要满足0 1 2 3.。。这样的如果中断了就无解,当然数很多是没关系的,那么我们就贪心的构造对于d【v】=d【u】+1来说,我们就建边u->v,这个就只是个模拟的过程,最后判断一下每个点的size有没有大于k就行了。

#include
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
vector<int>g[maxn],ans[maxn];
pair<int,int>d[maxn];
int sum=0,n,k,p[maxn];
set<int>s;
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i) scanf("%d",&d[i].first),d[i].second=i,s.insert(d[i].first); 
	sort(d+1,d+1+n);
	int root=d[1].second;
	for(int i=1;i<=n;++i)
	{
		if(d[i].first-d[i-1].first>=2) {puts("-1");return 0;}
		g[d[i].first].push_back(d[i].second);
	}
	if(d[2].first==0||d[1].first>0) {
		puts("-1");return 0;
	}
	auto it=s.begin();
	it++;
	while(it!=s.end())
	{
		auto tt=it;
		tt--;
		int t1=*it,t2=*(tt),m=0;
		for(int i:g[t1])
		{
			if(m==g[t2].size()) m=0;
			ans[g[t2][m]].push_back(i);
			m++;
		}
		it++;
	}
	for(int i=1;i<=n;++i)
	{
		if(i==root){
			if(ans[i].size()>k) {puts("-1");return 0;}
		}
		if(i!=root) if(ans[i].size()>=k) {puts("-1");return 0;}
		sum+=ans[i].size();
	}
	printf("%d\n",sum);
	for(int i=1;i<=n;++i)
	for(int j=0;j<ans[i].size();++j)
	printf("%d %d\n",i,ans[i][j]);
}

你可能感兴趣的:(贪心)