L2-025 分而治之（与L2-013 红色警戒 差不多一模一样）

原题链接

题目：

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

输入输出格式：

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

输入样例

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

代码与红色警戒基本一样，只有判断条件不同。

判断条件：当连通块数量等于所有城市数量时（即所有城市都是独立的，孤立无援的时候），就输出YES，else输出NO

可以参考 红色警戒AC代码

方法 一 dfs代码，大致思路就是对于每一个没被vis过的点进行深搜，如果这是一个lost后的城市就return，否则就一直深搜，把与这个点连通的所有点都vis一遍。

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
const int N=1e5+5;
vector <int> road[N];
bool vis[N];
bool lost[N];
int n,m,k,a,b,lostNum,t;
void dfs(int x)
{
    if (lost[x])return ;
    for (int i=0;i<road[x].size();i++)
    {
        if (lost[road[x][i]]) continue;
        if (!vis[road[x][i]]) vis[road[x][i]]=true,dfs(road[x][i]);
    }
}
int count()
{
    int cnt=0;
    for (int i=1;i<=n;i++) 
    {
        if (!vis[i]) {
            vis[i]=true;
            cnt++;
            dfs(i);
        }
    }
    return cnt;
}
int main()
{
    cin>>n>>m;
    for (int i=0;i<m;i++) 
    {
        cin>>a>>b;
        road[a].push_back(b);
        road[b].push_back(a);
    }
    cin>>k;
    int np;
    while (k--) 
    {
        lostNum=0;
        memset(vis,0,sizeof vis);
        memset(lost,0,sizeof lost);
        cin>>np;
        while (np--) {
            cin>>a;
            lost[a]=true;
        }
        t=count();
        if (t==n) cout<<"YES\n";
        else cout<<"NO\n";
    }
}

方法二 并查集

并查集的思路，先把输入的所有道路都存起来先，然后再对每个方案进行判断。如果一个方案能行的话，那么把所有丢失城市的lost设置为true后，其实一个unity都执行不了，这样f[i]==i的城市就会是n个

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+5;
int a,b,n,m,ans,k,np;
pair<int,int> road[N];
int f[N];
bool vis[N];
bool lost[N];
int find(int x)
{
    if (x!=f[x]) f[x]=find(f[x]);
    return f[x];
}
void unity(int x,int y)
{
    int xx=find(x);
    int yy=find(y);
    if (xx==yy) return ;
    f[xx]=yy;
}
int main()
{
    cin>>n>>m;
    //for (int i=1;i<=n;i++) f[i]=i;
    for (int i=0;i<m;i++) 
    {
        cin>>a>>b;
        road[i].first=a,road[i].second=b;
    }
    cin>>k;
    while (k-- ) 
    {
        ans=0;
        memset(lost,0,sizeof lost);
        for (int i=1;i<=n;i++) f[i]=i;
        cin>>np;
        while (np--)
        {
            cin>>a;
            lost[a]=true;
        }
        for (int i=0;i<m;i++) 
        {
            if (lost[road[i].first]||lost[road[i].second]) continue;
            unity(road[i].first,road[i].second);
        }
        for (int i=1;i<=n;i++) if (f[i]==i) ans++;//求当前有多少个连通块
        if (ans==n) cout<<"YES\n";
        else cout<<"NO\n";
    }
}