Codeforces Round #677 (Div. 3) D/1433D Districts Connection


https://codeforces.com/contest/1433/problem/D

找两个不同权值的节点A、B连起来,所有与A不同权值的连到A上,相同的连到B上。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#define debug(a) cout<<#a<<"="<>t;
  while(t--)
  {
  	LL n;cin>>n;
  	for(LL i=1;i<=n;i++) cin>>a[i];
  	LL idx=-1;
  	vector>pr;
  	for(LL i=2;i<=n;i++)
  	{
  		if(a[i]!=a[1]){
  			pr.push_back({1,i});
			idx=i;	
		}
	}
	if(idx==-1) {
		cout<<"NO"<