图上做dfs()
本题链接
题意: 给一张连通 的无向 图,同时给你两个重要点,询问图上有多少对点满足从一个点到另一个点都必须经过这两个重要点。
思路: 本题的答案应该就是只有b能到的点*只有a能到的点,因为图连通,一个小技巧,dfs时直接将a点标记走过,dfs(b)点,标记为0的点就是只有a点能到的点,dfs(a)亦然,快速秒杀这道题。
#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int N=200020;
int T,n,m,a,b,col[N],cnt1,cnt2;
vector<int>V[N];
void dfs(int x){
col[x]=true;
for(int i=0;i<V[x].size();i++){
int t=V[x][i];
if(col[t])continue;
dfs(t);
}
}
int main(){
scanf("%d",&T);
while(T--){
scanf("%d%d%d%d",&n,&m,&a,&b);
for(int i=1;i<=n;i++)V[i].clear();
for(int x,y,i=1;i<=m;i++)scanf("%d%d",&x,&y),V[x].pb(y),V[y].pb(x);
cnt1=cnt2=0;
col[a]=true,dfs(b);
for(int i=1;i<=n;i++){
printf("%d ",col[i]);
}
cout<<endl;
for(int i=1;i<=n;i++)
if(col[i]==0)cnt1++;
else col[i]=0;
col[b]=true,dfs(a);
for(int i=1;i<=n;i++)
if(col[i]==0)cnt2++;
else col[i]=0;
printf("%lld\n",1ll*cnt1*cnt2);
}
}