Description
Analysis
又是一道裸题,唉
双连通分量缩点之后在树上维护一些东西做LCA,怎么看都是练码力啊。
我好6啊3周连续3道连通分量类的题全部自信不对拍全部比赛时一次AC,23333
这说明自己的码力有所提升了,是好事
但是不对拍是不对的,还是要求稳,万一正式比赛翻车了就后悔莫及了。
Code
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
typedef long long ll;
const int N=200010,M=800010;
int n,m,tot,lyd,to[M],next[M],last[N];
int num,now,top,sta[N],low[N],dfn[N],bz[N],c[N];
int f[N][20],dep[N];
ll s1,s2,g[N][20],h[N][20],size[N];
int fa[N];
struct edge
{
int x,y;
}b[M];
bool vis[N],p[M];
void link(int u,int v)
{
to[++tot]=v,next[tot]=last[u],last[u]=tot;
}
void tarjan(int v)
{
sta[++top]=v;
low[v]=dfn[v]=++now;
bz[v]=lyd;
vis[v]=1;
efo(i,v)
{
if(p[i]) continue;
int u=to[i];
if(!dfn[u])
{
p[i]=p[i^1]=1;
tarjan(u);
p[i]=p[i^1]=0;
low[v]=min(low[v],low[u]);
}
else
if(vis[u]) low[v]=min(low[v],dfn[u]);
}
if(low[v]==dfn[v])
{
num++;
for(;sta[top+1]!=v;) c[sta[top--]]=num,size[num]++;
}
}
int getfa(int v)
{
if(!fa[v]) return v;
else return fa[v]=getfa(fa[v]);
}
void dfs(int v,int fr,int d)
{
dep[v]=d,f[v][0]=fr,g[v][0]=size[fr],h[v][0]=size[fr]*(size[fr]-1);
efo(i,v)
{
int u=to[i];
if(u==fr) continue;
dfs(u,v,d+1);
}
}
void jump(int &u,int i)
{
s1+=g[u][i],s2+=h[u][i],u=f[u][i];
}
ll lca(int u,int v)
{
s1=s2=0;
if(dep[u]<dep[v]) swap(u,v);
int x=u,y=v;
fd(i,log2(dep[u]),0)
if(dep[f[u][i]]>=dep[v]) jump(u,i);
if(u==v)
{
s1+=size[x],s2+=size[x]*(size[x]-1);
return s1*(s1-1)-s2;
}
fd(i,log2(dep[u]),0)
if(f[u][i]!=f[v][i]) jump(u,i),jump(v,i);
jump(u,0),jump(v,0);
s1+=size[x]+size[y]-size[u];
s2+=size[x]*(size[x]-1)+size[y]*(size[y]-1)-size[u]*(size[u]-1);
return s1*(s1-1)-s2;
}
int main()
{
int _,u,v;
ll ans=0;
scanf("%d %d %d",&n,&m,&_);
tot=1;
fo(i,1,m)
{
scanf("%d %d",&u,&v);
b[i].x=u,b[i].y=v;
link(u,v),link(v,u);
}
fo(i,1,n)
if(!bz[i])
{
while(top) sta[top--]=0;
++lyd;
tarjan(i);
}
tot=0;
memset(last,0,sizeof(last));
fo(i,1,m)
{
u=c[b[i].x],v=c[b[i].y];
if(u==v) continue;
int fu=getfa(u),fv=getfa(v);
if(fu==fv) continue;
fa[v]=u;
link(u,v),link(v,u);
}
dfs(1,0,1);
fo(j,1,log2(n))
fo(i,1,n)
{
f[i][j]=f[f[i][j-1]][j-1];
g[i][j]=g[i][j-1]+g[f[i][j-1]][j-1];
h[i][j]=h[i][j-1]+h[f[i][j-1]][j-1];
}
while(_--)
{
scanf("%d %d",&u,&v);
u=c[u],v=c[v];
if(u==v) continue;
ans+=lca(u,v);
}
printf("%lld",ans);
return 0;
}