L2-025 分而治之

1.题目

L2-025 分而治之
分数 25

全屏浏览

切换布局
作者 陈越
单位 浙江大学
分而治之，各个击破是兵家常用的策略之一。在战争中，我们希望首先攻下敌方的部分城市，使其剩余的城市变成孤立无援，然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序，判断每个方案的可行性。

输入格式：
输入在第一行给出两个正整数 N 和 M（均不超过10 000），分别为敌方城市个数（于是默认城市从 1 到 N 编号）和连接两城市的通路条数。随后 M 行，每行给出一条通路所连接的两个城市的编号，其间以一个空格分隔。在城市信息之后给出参谋部的系列方案，即一个正整数 K （≤ 100）和随后的 K 行方案，每行按以下格式给出：

Np v[1] v[2] ... v[Np]
其中 Np 是该方案中计划攻下的城市数量，后面的系列 v[i] 是计划攻下的城市编号。

输出格式：
对每一套方案，如果可行就输出YES，否则输出NO。

输入样例：
10 11
8 7
6 8
4 5
8 4
8 1
1 2
1 4
9 8
9 1
1 10
2 4
5
4 10 3 8 4
6 6 1 7 5 4 9
3 1 8 4
2 2 8
7 9 8 7 6 5 4 2
输出样例：
NO
YES
YES
NO
NO

2.分析

        用判断入度或用并查集或深度搜索都可以。

        这里用并查集。

3.代码

#include<iostream>
#include<vector>
using namespace std;
const int MAX=10010;         //必须比10000大
vector<int> v[MAX];          //存储结点之间的边
int N,M,K;
int vis[MAX],fa[MAX];
void init(){                  //并查集初始化函数
    for(int i=1;i<=N;i++){fa[i]=i;}
} 
int find(int x){              //并查集查找函数
    if(fa[x]==x) return x;
    else{
        fa[x]=find(fa[x]);
        return fa[x];
    }
}
void add(int a,int b){        //并查集合并函数
    int f_a=find(a);
    int f_b=find(b);
    fa[f_a]=f_b;
}
void clear(){                //重置标记数组
    for(int i=1;i<=N;i++){
        vis[i]=0;
    }
}
void build(){              //建立并查集
    for(int i=1;i<=N;i++){
        for(int j=0;j<v[i].size();j++){
            if(!vis[v[i][j]]&&!vis[i])
                add(i,v[i][j]);
        }
    }
}
bool check(){              //判断方案的可行性
    for(int i=1;i<=N;i++){
        if(find(i)!=i)
            return false;
    }
    return true;
}
int main(){
    cin>>N>>M;
    int x,y;
    for(int i=0;i<M;i++){
        cin>>x>>y;
        v[x].push_back(y);     //记录边
        v[y].push_back(x);
    }
    cin>>K;
    int uns,id;
    for(int i=0;i<K;i++){
        cin>>uns;
        clear();             //重置标记数组
        for(int j=0;j<uns;j++){
            cin>>id;
            vis[id]++;           //记录失去的城市
            //cout<<id<<" "<<vis[id]<<endl;
        }
        init();           //初始化并查集
        build();           //建立并查集
        if(check()) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}