[BZOJ5463] [APIO2018] 铁人两项

题目链接

LOJ.

BZOJ.

Solution

先建圆方树。

我们考虑暴力，枚举一个点对，我们枚举的点都是圆点，然后统计中间那个点可以取的位置的数量，加起来就是答案。

那么怎么统计呢，我们对于每个点赋一个点权，方点点权为点双的大小，圆点点权为\(-1\)。

那么这条路径的点权和就是答案，注意要统计到端点的权值。

然后优化就很显然了，直接枚举每个点被算了多少次就行了，这个随便算一下就好了。

复杂度\(O(n)\)。

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

#define int long long 

void read(int &x) {
    x=0;int f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f;
}
 
void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(!x) return ;print(x/10),putchar(x%10+48);
}
void write(int x) {if(!x) putchar('0');else print(x);putchar('\n');}

#define lf double
#define ll long long 

const int maxn = 4e5+10;
const int inf = 1e9;
const lf eps = 1e-8;

int cnt,n,m,val[maxn],ans,rt;

struct Tree {
	int head[maxn],tot,vis[maxn],sz[maxn];
	struct edge{int to,nxt;}e[maxn<<1];

	void add(int u,int v) {e[++tot]=(edge){v,head[u]},head[u]=tot;}
	void ins(int u,int v) {add(u,v),add(v,u);}

	void dfs(int x,int fa) {
		vis[x]=1,sz[x]=x<=n;
		for(int i=head[x];i;i=e[i].nxt) if(e[i].to!=fa) dfs(e[i].to,x),sz[x]+=sz[e[i].to];
	}

	void solve(int x,int fa) {
		int res=0,s=0;
		for(int i=head[x];i;i=e[i].nxt)
			if(e[i].to!=fa) res+=s*sz[e[i].to],s+=sz[e[i].to];
		res+=s*(sz[rt]-sz[x]);res<<=1;ans+=res*val[x];
		for(int i=head[x];i;i=e[i].nxt) if(e[i].to!=fa) solve(e[i].to,x);
	}
}T;

struct Graph {
	int head[maxn],tot,dfn[maxn],low[maxn],dfn_cnt,sta[maxn],top;
	struct edge{int to,nxt;}e[maxn<<1];

	void add(int u,int v) {e[++tot]=(edge){v,head[u]},head[u]=tot;}
	void ins(int u,int v) {add(u,v),add(v,u);}

	void tarjan(int x,int fa) {
		dfn[x]=low[x]=++dfn_cnt,sta[++top]=x;
		for(int v,i=head[x];i;i=e[i].nxt) {
			if((v=e[i].to)==fa) continue;
			if(!dfn[v]) tarjan(v,x),low[x]=min(low[x],low[v]);
			else {low[x]=min(low[x],dfn[v]);continue;}
			if(low[v]>=dfn[x]) {
				++cnt;T.ins(cnt,x);val[cnt]++;
				while(top) {
					int now=sta[top--];T.ins(now,cnt),val[cnt]++;
					if(now==v) break;
				}
			}
		}
	}
}G;

signed main() {
	read(n),read(m);for(int i=1,x,y;i<=m;i++) read(x),read(y),G.ins(x,y);
	cnt=n;for(int i=1;i<=n;i++) if(!G.dfn[i]) G.tarjan(i,0);
	for(int i=1;i<=n;i++) val[i]=-1;
	for(int i=1;i<=n;i++) if(!T.vis[i]) rt=i,T.dfs(i,0),T.solve(i,0),ans-=T.sz[i]*(T.sz[i]-1)*2;
	write(ans);
	return 0;
}