codeforces 776D The Door Problem(带权并查集)

版权声明:本文为博主原创文章,未经博主允许不得转载。如有错误,欢迎指出~(@^_^@)~ https://blog.csdn.net/zwj1452267376/article/details/57082351
D. The Door Problem
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

Moriarty has trapped n people in n distinct rooms in a hotel. Some rooms are locked, others are unlocked. But, there is a condition that the people in the hotel can only escape when all the doors are unlocked at the same time. There are m switches. Each switch control doors of some rooms, but each door is controlled by exactly two switches.

You are given the initial configuration of the doors. Toggling any switch, that is, turning it ON when it is OFF, or turning it OFF when it is ON, toggles the condition of the doors that this switch controls. Say, we toggled switch 1, which was connected to room 12 and 3 which were respectively locked, unlocked and unlocked. Then, after toggling the switch, they become unlocked, locked and locked.

You need to tell Sherlock, if there exists a way to unlock all doors at the same time.

Input

First line of input contains two integers n and m (2 ≤ n ≤ 1052 ≤ m ≤ 105) — the number of rooms and the number of switches.

Next line contains n space-separated integers r1, r2, ..., rn (0 ≤ ri ≤ 1) which tell the status of room doors. The i-th room is locked if ri = 0, otherwise it is unlocked.

The i-th of next m lines contains an integer xi (0 ≤ xi ≤ n) followed by xi distinct integers separated by space, denoting the number of rooms controlled by the i-th switch followed by the room numbers that this switch controls. It is guaranteed that the room numbers are in the range from 1 to n. It is guaranteed that each door is controlled by exactly two switches.

Output

Output "YES" without quotes, if it is possible to open all doors at the same time, otherwise output "NO" without quotes.

Examples
input
3 3
1 0 1
2 1 3
2 1 2
2 2 3
output
NO
input
3 3
1 0 1
3 1 2 3
1 2
2 1 3
output
YES
input
3 3
1 0 1
3 1 2 3
2 1 2
1 3
output
NO
Note

In the second example input, the initial statuses of the doors are [1, 0, 1] (0 means locked, 1 — unlocked).

After toggling switch 3, we get [0, 0, 0] that means all doors are locked.

Then, after toggling switch 1, we get [1, 1, 1] that means all doors are unlocked.

It can be seen that for the first and for the third example inputs it is not possible to make all doors unlocked.


虽然大佬们说CF题目质量严重下降,屡屡出现原题,但是我还是不会做(去屎吧,愚蠢的纯真姬


注意:每一个门都是受两个开关限制的。

代码如下:


#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm> 
using namespace std;
const int maxn = 1e5+10;
typedef vector<int> vec;
int door[maxn];
int tree[2*maxn];
vec G[maxn];

int find(int x)
{
	if(x==tree[x])
		return x;
	return tree[x]=find(tree[x]);
}

void merge(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
		tree[fx]=fy;
}


int main()
{
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=1;i<=n;++i){
			scanf("%d",&door[i]);
			G[i].clear();
		}
			
		for(int i=1;i<=2*m;++i)
			tree[i]=i;
		int k,x;
		for(int i=1;i<=m;++i)
		{
			scanf("%d",&k);
			while(k--)
			{
				scanf("%d",&x);
				G[x].push_back(i);
			}
		}
		for(int i=1;i<=n;++i)
		{
			int x=G[i][0];
			int y=G[i][1];
			if(door[i]){
				merge(x,y);
				merge(x+m,y+m);
			}
			else{
				merge(x,y+m);
				merge(x+m,y);
			}
		}
		int flag=1;
		for(int i=1;i<=m;++i)
		{
			if(find(i)==find(i+m))
			{
				flag=0;
				break;
			}
		}
		if(flag)
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
} 


猜你喜欢

转载自blog.csdn.net/zwj1452267376/article/details/57082351
今日推荐