Codeforces Round #562 (Div. 2) B. Pair(思维)

 

题目链接：https://codeforces.com/contest/1169/problem/B

       题意是输入n和m，然后输入m组数，输入的数都为1-n的数，然后问能否从1-n中找到两个不相同的数x和y使得这m组数中，每组都至少有一个数等于x或者等于y，如果可以输出YES，否则输出NO。

       思路大致就是暴力，但是最简单的暴力的时间复杂度应该是n^2*m的，所以需要换一种思路，用ma来存每个数出现的次数，我们枚举x和y中的x，然后看有多少组数中含有x，然后我们去寻找出现了m-ma[x]次数的数，大致的思路就是这样，然后需要处理一个容斥的问题，就是我们在找x和y的时候看能否覆盖所有的m组数的时候，应该是判断ma[x] + ma[y] - ma[x和y一起出现] == m

---

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> p;
int n,m;
vector<int> v[300005];

int main()
{
	scanf("%d%d",&n,&m);
	map<int,int> ma;
	map<pair<int,int>,int> maa;
	int Max = 0;
	for(int i=0;i<m;i++){
		int x, y;
		scanf("%d%d",&x,&y);
		maa[p(x, y)] ++;
		maa[p(y, x)] ++;
		ma[x] ++;
		ma[y] ++;
		Max = max(Max, max(ma[x], ma[y]));
	}
	for(int i=1;i<=n;i++){
		v[ma[i]].push_back(i);
	}
	bool flag = false;
	for(int i=1;i<=n;i++){
		int xx = ma[i];
		for(int j=m-xx;j<=Max;j++){
			for(int k=0;k<v[j].size();k++){
				if(v[j][k] == i) continue;
				if(xx + ma[v[j][k]] - maa[p(i, v[j][k])] == m){
					flag = true;
					break;
				}
			}
			if(flag) break;
		}
		if(flag) break;
	}
	if(flag) puts("YES");
	else puts("NO");
	return 0;
}