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
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 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;
}

 

你可能感兴趣的:(dfs树)