【Codeforces 1433 D】Districts Connection,构造树,脑筋急转弯

举报
小哈里 发表于 2022/05/10 22:53:29 2022/05/10
【摘要】 problem D. Districts Connection time limit per test1 second memory limit per test256 megabytes inputs...

problem

D. Districts Connection
time limit per test1 second
memory limit per test256 megabytes
inputstandard input
outputstandard output
There are n districts in the town, the i-th district belongs to the ai-th bandit gang. Initially, no districts are connected to each other.

You are the mayor of the city and want to build n−1 two-way roads to connect all districts (two districts can be connected directly or through other connected districts).

If two districts belonging to the same gang are connected directly with a road, this gang will revolt.

You don’t want this so your task is to build n−1 two-way roads in such a way that all districts are reachable from each other (possibly, using intermediate districts) and each pair of directly connected districts belong to different gangs, or determine that it is impossible to build n−1 roads to satisfy all the conditions.

You have to answer t independent test cases.

Input
The first line of the input contains one integer t (1≤t≤500) — the number of test cases. Then t test cases follow.

The first line of the test case contains one integer n (2≤n≤5000) — the number of districts. The second line of the test case contains n integers a1,a2,…,an (1≤ai≤109), where ai is the gang the i-th district belongs to.

It is guaranteed that the sum of n does not exceed 5000 (∑n≤5000).

Output
For each test case, print:

NO on the only line if it is impossible to connect all districts satisfying the conditions from the problem statement.
YES on the first line and n−1 roads on the next n−1 lines. Each road should be presented as a pair of integers xi and yi (1≤xi,yi≤n;xi≠yi), where xi and yi are two districts the i-th road connects.
For each road i, the condition a[xi]≠a[yi] should be satisfied. Also, all districts should be reachable from each other (possibly, using intermediate districts).

Example
inputCopy
4
5
1 2 2 1 3
3
1 1 1
4
1 1000 101 1000
4
1 2 3 4
outputCopy
YES
1 3
3 5
5 4
1 2
NO
YES
1 2
2 3
3 4
YES
1 2
1 3
1 4

solution

/*
题意:
+ 用n-1条边将n(<5e6)个点连成一棵树,满足相邻的点权不同,输出可行方案
思路:
+ 脑筋急转弯()
+ 考虑特解:首先随便选个点a1,把所有ai!=a1都和a1连起来,于是剩下的都==a1,然后把他们都和a1上随便找的子节点连起来就行。当且仅当所有点权相同才会无解。
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 1e5+10;
int a[maxn];
int main(){
	int T;  cin>>T;
	while(T--){
		int n;  cin>>n;
		for(int i = 1; i <= n; i++)
			cin>>a[i];
		
		vector<pair<int,int> >ans;
		for(int i = 2; i <= n; i++){
			if(a[i]!=a[1])ans.push_back({1,i});
		}
		
		if(ans.size()==0){
			cout<<"NO\n"; continue;
		}
		for(int i = 2; i <= n; i++){
			if(a[i]==a[1])ans.push_back({ans[0].second,i});
		}
		cout<<"YES\n";
		for(auto [x,y] : ans)cout<<x<<" "<<y<<"\n";
	}
	return 0;
}


文章来源: gwj1314.blog.csdn.net,作者:小哈里,版权归原作者所有,如需转载,请联系作者。

原文链接:gwj1314.blog.csdn.net/article/details/113276559

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

举报
请填写举报理由
0/200