刷题 day 2:疫情防控

疫情尚未结束，严防疫情反复。为了做好疫情防控工作，国内设置了地区风险等级，对于中高风险地区的人员采取限制移动、居家隔离等手段。

为了研究疫情防控对于跨地区交通运输的影响，假设现在有 N 个机场，M 条航线，每天都会新增一个防控地区，一个防控地区会导致一个机场无法正常运作，航线也自然无法正常运行，每天会有 Qi​ 对旅客从 Xi​ 机场前往 Yi​ 机场，请计算有多少对旅客会受到影响无法完成行程。

旅客只要能直达或通过若干次中转，且乘坐的所有航线的出发和到达机场都正常运作，即视作可完成行程。

输入格式:

输入第一行是三个整数 N,M,D (1≤N≤5×104, 1≤M≤2×105, 1≤D≤103), 表示机场数、航线数以及新增防控地区的天数。

接下来首先有 M 行，每行给出空格分隔的两个数字 A 和 B，表示编号为 A 和 B 的机场之间有一条航线。航线是双向的，机场编号从 1 到 N。

然后是 D 块输入，每块输入内第一行为空格分隔的两个整数 C 和 Q （1≤Q≤103)，表示新增机场编号为 C 所在的城市为防控地区，今天有 Q 段行程。数据保证新增的城市之前一定不是防控地区。

接下来的 Q 行，每行是空格分隔的两个数字 X 和 Y，表示编号为 X 和 Y 的机场的一段行程。行程有可能包括之前就已经成为防控地区的城市。

输出格式:

对于每天的询问，请在一行中输出在新增了一个防控地区后当天的行程有多少不能成行。

输入样例:

5 5 3
1 2
1 3
1 5
2 5
3 4
4 3
1 3
1 4
2 3
5 3
3 4
2 3
3 5
1 3
2 3
2 5
3 4

输出样例:

1
2
3

如果先建立并查集，再根据要求进行删除重建，运行太耗时，可以通过回溯，从最后一天开始加上被删除的点，

#include<bits/stdc++.h>
using namespace std;
const int N= 1e5+10;
//并查集，倒着进行，将没有被删除的机场依次添加上，并判断当天Q对行程是否可行

int p[N],n,m,d,r[N];
//初始化
int init(int n){
    for(int i=1;i<=n;i++){
        p[i]=i;
        r[i]=1;
    }
    return 0;

}
//定义并查集的查找，添加
int find(int x){
    if(x==p[x]) return p[x];
    else find(p[x]);
}

int add(int x,int y){
    int x1=p[x],y1=p[y];
    if(x1!=y1){
        if(r[x1]>r[y1]) p[y1]=x1;
        else{
            p[x1]=y1;
            if(r[x1]==r[y1]) r[y1]++;
        }
    }
    return 0;
}

vector<int> g[N];//储存航线
int vis[N],dd[N];//记录删除的点
vector<pair<int,int>> a[N];//每天的Q对行程
int ant[N];//不能通过的行程

int main(){
    cin>>n>>m>>d;
    while(m--){
        int a,b;
        cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);

    }
    for(int i=1;i<=d;++i){
        int c,q;
        cin>>c>>q;
        vis[c]=1;
        dd[i]=c;
        while(q--){
            int x,y;
            cin>>x>>y;
            a[i].push_back({x,y});

        }
    }

    init(n);
    //没有被删除的点，连接；
    for(int i=1;i<=n;i++){
        if(vis[i]==0){
            for(int j=0;j<g[i].size();j++){
                if(vis[g[i][j]]==0 ) add(i,g[i][j]);
            }
        }
    }

    for(int i=d;i>0;i--){
         //记录每天不能到达的行程次数
        for(int j=0;j<a[i].size();j++){
            int x1=find(a[i][j].first),y1=find(a[i][j].second);
            if(x1!=y1) ant[i]++;
        }
        //回溯添加当天删除的点；
        vis[dd[i]]=0;
        for(int j=0;j<g[dd[i]].size();++j){
            if(vis[g[dd[i]][j]]==0)add(dd[i],g[dd[i]][j]);
        }


    }
    for(int i=1;i<=d;i++){
        cout<<ant[i]<<endl;
    }


}