[POI2013]Polaryzacja

[POI2013]Polaryzacja

题目大意：

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

思路：

显然最小可达点对数是\(n-1\)。一种构造就是根结点全是入边，与根结点相邻的点全是出边……以此类推。最后相邻的点对会被统计一次，其余的均不会被统计。

对于最大可达点对数，一定存在一种方案，使得树根是树的任意一个重心时，将所有子树分成两部分，一部分有\(k\)个点，另一部分有\((n-k-1)\)个点，答案就是\(\max\{(n-k-1)k+\sum(dep(i)-1)\}\)。

具体证明略。

我们可以用一个背包来求出所有可能的\(k\)，但是时间复杂度是\(\mathcal O(n^2)\)，就算使用bitset优化也无济于事。

因此我们可以将所有子树\(size\)按照大小分开转移。\(>\sqrt n\)的不超过\(\sqrt n\)个，可以直接暴力转移；\(\le\sqrt n\)的按照大小分组一起转移，具体转移时可以按照二进制拆分。

时间复杂度\(\mathcal O(\frac{n\sqrt n}\omega)\)。

源代码：

#include<cmath>
#include<cstdio>
#include<cctype>
#include<vector>
#include<bitset>
#include<algorithm>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
typedef long long int64;
const int N=250001,M=501;
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
    e[u].push_back(v);
    e[v].push_back(u);
}
int64 sum;
int n,m,size[N],max[N],cen,cnt[M];
std::bitset<N> f;
void dfs(const int &x,const int &par) {
    size[x]=1;
    for(unsigned i=0;i<e[x].size();i++) {
        const int &y=e[x][i];
        if(y==par) continue;
        dfs(y,x);
        size[x]+=size[y];
        max[x]=std::max(max[x],size[y]);
    }
    max[x]=std::max(max[x],n-size[x]);
    if(max[x]<max[cen]) cen=x;
}
void dfs(const int &x,const int &par,const int &dep) {
    sum+=dep-1;
    for(unsigned i=0;i<e[x].size();i++) {
        const int &y=e[x][i];
        if(y==par) continue;
        dfs(y,x,dep+1);
    }
}
int main() {
    max[0]=n=getint(),m=sqrt(n);
    for(register int i=1;i<n;i++) {
        add_edge(getint(),getint());
    }
    dfs(1,0);
    dfs(cen,0,1);
    f[0]=true;
    for(register unsigned i=0;i<e[cen].size();i++) {
        const int &y=e[cen][i];
        if(size[y]<=m) {
            cnt[size[y]]++;
        } else {
            f|=f<<size[y];
        }
    }
    for(register int i=1;i<=m;i++) {
        for(register int j=1;j<=cnt[i];j<<=1) {
            f|=f<<(i*j);
            cnt[i]-=j;
        }
        if(cnt[i]) f|=f<<(i*cnt[i]);
    }
    int k;
    int64 ans=0;
    for(k=0;k<=n;k++) {
        if(f[k]) ans=std::max(ans,sum+(int64)k*(n-k-1));
    }
    printf("%d %lld\n",n-1,ans);
    return 0;
}

转载于:https://www.cnblogs.com/skylee03/p/9608312.html