【SCAU 新生赛】18247 aler的旅游计划 并查集模板题

18247 aler的旅游计划
该题有题解

时间限制:1000MS 代码长度限制:10KB
提交次数:15 通过次数:0 收入:10

题型: 编程题 语言: 不限定
Description
aler想要去旅游，他查了下地图发现，一共有n个城市，这n个城市间又有m条道路，道路是双向的。
每个城市都会有一个通行证，如果没有通行证就不能经过这个城市。aler只有一些城市的通行证，
他现在有q个询问，每个询问会给出两个整数a和b，他想要知道他从城市a能不能走到城市b，
如果可以，输出“YES”，不能则输出“NO”（不包含双引号）

输入格式
第一行输入两个数字n和m，分别代表n个城市和m条道路。第二行输入n个数字。
数字只会是0或者1，第i个数字为0说明aler没有第i个城市的通行证，为1说明有该城市的通行证。
接下来m行输出两个数字u和v，代表城市u和城市v之间有一条道路。
再输入一个数字q，代表询问的次数。接下来q行，每行都有两个数字a和b，代表询问的城市a和城市b。
(2 <= n <= 10000, 0 <= m <= 100000, 1 <= u, v, a, b <= n，1 <= q <= 100000)

输出格式
输出q行，每行输出城市a能不能走到城市b，如果可以，输出“YES”，不能则输出“NO”（不包含双引号）

输入样例
6 6
1 0 1 1 0 1
1 2
1 3
2 4
3 4
1 5
5 6
2
1 6
1 4

输出样例
NO
YES

提示
u可能等于v，a也可能等于b，也可能会有重边。
注意如果aler没有城市a或城市b的通行证的话，那他也无法通行。
当然城市a是可以到达a城市的，不管aler有没有a城市的通行证。

样例解释：从城市1走到城市6的话必须通过5，而aler没有城市5的通行证，所以不能经过。从城市1走到城市4可以直接走1-3-4到达。

题意：给出哪些点能到哪些不能到，和m条边，然后询问某两个点之间有无路可通

思路：

二星给多了。。。并查集模板题，给的n个数用来判断两个点能不能合并，然后对m条边进行并查集。询问的时候看看是否根节点相同即可。

AC代码：

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
const int maxn = 1e5+200;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'|ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };

ll a[maxn];
ll fa[maxn];
ll n,m;

void init()
{
    rep(i,1,n) fa[i] = i;
}

ll get(ll x)
{
    return fa[x] = fa[x] == x? x : get(fa[x]);
}

void merge(ll x, ll y)
{
    ll a = get(x),  b = get(y);
    if(a!=b)
    {
        fa[b] = a;
    }
}

int main()
{
    while(cin>>n>>m)
    {
        init();
        rep(i,1,n) a[i] = read();
        rep(i,1,m)
        {
            ll x, y;
            x  = read(), y  = read();
            if(a[x]&&a[y])
            merge(x,y);
        }
        ll q; q = read();
        rep(i,1,q)
        {
            ll x = read(), y = read();
            if(get(x)==get(y)) cout<<"YES"<<'\n';
            else cout<<"NO"<<'\n';
        }
    }
    return 0;
}