【BZOJ3425】Poi2013 Polarization 猜结论+DP

【BZOJ3425】Poi2013 Polarization

Description

给定一棵树，可以对每条边定向成一个有向图，这张有向图的可达点对数为树上有路径从u到达v的点对(u,v)个数。求最小可达点对数和最大可达点对数

n<=250000

Sample Input

4
1 2
1 3
1 4

Sample Output

3 5

题解：想了一晚上，怎么想怎么是个搭建双塔，结果看题解发现还真tm是搭建双塔。

本题的结论有点神，不过很好猜，证明见Claris博客。

第一问的答案一定是n-1，因为树是个二分图，所以我们将树黑白染色后所有边都从黑点指向白点即可。

第二问的答案是什么呢？我们猜测一定是先选中一个点，然后一半的边都指向这个点，另一半的边从这个点指出去。而选择哪个点呢？显然是重心啦。然后我们将重心之外所有联通块的大小都拿出来，这就变成了搭建双塔问题。

话说搭建双塔不是$O(n^2)$的吗？n=250000你逗我？然而正解很奇特。

当联通块大小<sqrt(n)时，我们将所有这样的联通块合到一起跑分组背包；当联通块大小>sqrt(n)时，这样的联通块不超过sqrt(n)个，所以暴力DP即可。DP时用bitset维护，于是时间复杂度就变成了神奇的$O({n \sqrt{n}\over 32})$。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <bitset>
#include <cmath>
using namespace std;
const int maxn=250010;
typedef long long ll;
int B,n,m,cnt,rt,mn;
ll sum,ans;
int to[maxn<<1],next[maxn<<1],head[maxn],siz[maxn],v[maxn],dep[maxn],s[maxn];
bitset<maxn> f;
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')	f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+(gc^'0'),gc=getchar();
	return ret*f;
}
inline void add(int a,int b)
{
	to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
}
void dfs(int x)
{
	siz[x]=1,sum+=dep[x]-1;
	int i,tmp=0;
	for(i=head[x];i!=-1;i=next[i])	if(!dep[to[i]])
		dep[to[i]]=dep[x]+1,dfs(to[i]),siz[x]+=siz[to[i]],tmp=max(tmp,siz[to[i]]);
	tmp=max(tmp,n-siz[x]);
	if(tmp<mn)	mn=tmp,rt=x;
}
int main()
{
	n=rd(),B=int(sqrt(double(n))),mn=1<<30;
	memset(head,-1,sizeof(head));
	int i,j,a,b;
	for(i=1;i<n;i++)	a=rd(),b=rd(),add(a,b),add(b,a);
	dep[1]=1,dfs(1),memset(dep,0,sizeof(dep));
	sum=0,dep[rt]=1,dfs(rt);
	for(i=head[rt];i!=-1;i=next[i])	v[++m]=siz[to[i]];
	f[0]=1;
	for(i=1;i<=m;i++)
	{
		if(v[i]<=B)	s[v[i]]++;
		else	f=f|(f<<v[i]);
	}
	for(i=1;i<=B;i++)
	{
		for(j=1;j<=s[i];s[i]-=j,j<<=1)	f=f|(f<<(i*j));
		if(s[i])	f=f|(f<<(i*s[i]));
	}
	for(i=0;i<=n;i++)	if(f[i])	ans=max(ans,sum+(ll)i*(n-1-i));
	printf("%d %lld",n-1,ans);
	return 0;
}