codeforces1277E Two Fairs

https://codeforces.com/contest/1277/problem/E

這題直接搞的話,先邊雙縮點,然後記錄每個邊雙直接連的邊是連的哪兩個點,然後再考慮加不a,b是不是剛好是它所在邊雙的出口,再計算。這樣寫估計得200+行

然而這題用dfs樹就會簡單許多,直接以a爲起點開始跑tarjan,如果跑到b,b的某個兒子v,low[v]>=dfn[b],那麼說明v所在的子樹所有點到a都需要經過b,於是sum1+=son[v],然後再以b爲起點做一遍一樣的動作

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;

int n,m,cas,k,cnt,tot,a,b,ind;
int sum1,sum2;
int dfn[maxl],low[maxl],son[maxl];
char s[maxl];
bool in[maxl]; 
ll ans;
vector<int> e[maxl];

inline void prework()
{
	scanf("%d%d%d%d",&n,&m,&a,&b);
	int u,v;
	for(int i=1;i<=n;i++)
		e[i].clear();
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
	}
} 

inline void tarjan(int u,int fa,int &b,int &sum)
{
	dfn[u]=low[u]=++ind;son[u]=1;
	for(int v:e[u])
	{
		if(!dfn[v])
		{
			tarjan(v,u,b,sum);
			son[u]+=son[v];
			low[u]=min(low[u],low[v]);
			if(u==b && low[v]>=dfn[u])
				sum+=son[v]; 
		}
		else if(v!=fa)
			low[u]=min(low[u],dfn[v]);
	}
}

inline void mainwork()
{
	for(int i=1;i<=n;i++)
		son[i]=0,dfn[i]=0;
	sum1=sum2=0;ind=0;
	tarjan(a,0,b,sum1);
	for(int i=1;i<=n;i++)
		dfn[i]=0,son[i]=0;
	ind=0;
	tarjan(b,0,a,sum2);
	ans=1ll*sum1*sum2;
}

inline void print()
{
	printf("%lld\n",ans);
}

int main()
{
	int t=1;
	scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章