hihocoder 1247 树形XX~~

hihocoder 1247

描述

王胖浩有一个n个节点的树。他有特殊的能力,能将树划分成若干条链。

一个有根树的树链剖分为将整棵树划分成若干条从一个点到这个点的祖先(包括本身)的链,并且这些链没有公共点。如果一条边在某条链上,即这条边的两个端点在同一个链上,那么这个边为重边,否则为轻边。

一个树链剖分的代价为所有点对的路径上轻边的个数之和。这里将u,v和v,u当成同一个点对。一个有根树的最优树链剖分为所有方案中代价最小的方式。

他想通过合理地使用他的特殊能力,使得树链剖分最优。

他得到了一个无根树,他想知道以每个点为根对应的有根树的最优树链剖分的代价。



每个边的代价为两头节点数的乘积

那么对于一个确定了根的树,非叶子节点将选择一条代价最大的边作为重边,贪心即可得到最小的代价

对于每一个点为根的答案,不可能暴力求解,需要进行信息维护。

一、先以1为根,贪心求出这个树的最小代价,第一次DFS

二、把根转移到孩子节点,把外界的信息传递给孩子节点,同时修正以孩子节点为根的树的结构~~第二次DFS

这里的外界信息指得是除去当前孩子节点为根的子树的其他子树的信息


总的来说这类题目就是先求出一个根的答案(第一个DFS),然后将根往孩子节点移动,传递外界信息给孩子节点以及修正新树的结构(第二次DFS)

#include 
#include 
#include
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define exp 1e-8
#define ll long long
#define INF 0x3f3f3f3f3f3f3f3f
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define all(a) a.begin(),a.end()
#define mm(a,b) memset(a,b,sizeof(a));
#define for0(a,b) for(int a=0;a<=b;a++)//0---(b-1)
#define for1(a,b) for(int a=1;a<=b;a++)//1---(b)
#define rep(a,b,c) for(int a=b;a<=c;a++)//b---c
#define repp(a,b,c)for(int a=b;a>=c;a--)///
#define cnt_one(i) __builtin_popcount(i)
#define stl(c,itr) for(__typeof((c).begin()) itr=(c).begin();itr!=(c).end();itr++)
using namespace std;
void bug(string m="here"){cout< inline void READ(__ll &m){__ll x=0,f=1;char ch=getchar();while(!(ch>='0'&&ch<='9')){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}m=x*f;}
templateinline void read(__ll &m){READ(m);}
templateinline void read(__ll &m,__ll &a){READ(m);READ(a);}
templateinline void read(__ll &m,__ll &a,__ll &b){READ(m);READ(a);READ(b);}
templateinline void read(__ll &m,__ll &a,__ll &b,__ll &c){READ(m);READ(a);READ(b);READ(c);}
templateinline void read(__ll &m,__ll &a,__ll &b,__ll &c,__ll &d){READ(m);READ(a);READ(b);READ(c);read(d);}
template < class T > inline  void out(T a){if(a<0){putchar('-');a=-a;}if(a>9)out(a/10);putchar(a%10+'0');}
template < class T > inline  void outln(T a){out(a);puts("");}
template < class T > inline  void out(T a,T b){out(a);putchar(' ');out(b);}
template < class T > inline  void outln(T a,T b){out(a);putchar(' ');outln(b);}
template < class T > inline  void out(T a,T b,T c){out(a);putchar(' ');out(b);putchar(' ');out(c);}
template < class T > inline  void outln(T a,T b,T c){out(a);putchar(' ');outln(b);putchar(' ');outln(b);}
template < class T > T gcd(T a, T b) { return b ? gcd(b, a % b) : a; }
template < class T > T lcm(T a, T b) { return a / gcd(a, b) * b; }
template < class T > inline void rmin(T &a, const T &b) { if(a > b) a = b; }
template < class T > inline void rmax(T &a, const T &b) { if(a < b) a = b; }
template < class T > T pow(T a, T b) { T r = 1; while(b > 0) { if(b & 1) r = r * a; a = a * a; b /= 2; } return r; }
template < class T > T pow(T a, T b, T mod) { T r = 1; while(b > 0) { if(b & 1) r = r * a % mod; a = a * a % mod; b /= 2; } return r; }

const int cnt_edge=200100;  //修改啊
const int cnt_v=100100;
int head[cnt_v],cnt_e;
struct EDGE{int u,v,next,cost;}edge[cnt_edge];
void addedge(int u,int v,int cost=0)
{edge[cnt_e].u=u;edge[cnt_e].v=v;edge[cnt_e].cost=cost;edge[cnt_e].next=head[u];head[u]=cnt_e++;}
void init(){cnt_e=0;memset(head,-1,sizeof(head));}
#define erg(i,u) for(int i=head[u];i!=-1;i=edge[i].next)

ll n,m,fi[cnt_v*2],se[cnt_v*2],size[cnt_v*2];
ll d[cnt_v],tot[cnt_v],pre[cnt_v],ans[cnt_v];
void dfs1(int u,int fa)
{
    erg(i,u)
    {
        int v=edge[i].v;
        if(v==fa)continue;
        dfs1(v,u);
        d[u]+=d[v];
        size[u]+=size[v];
        if(size[v]*(n-size[v])>size[fi[u]]*(n-size[fi[u]]))
            se[u]=fi[u],fi[u]=v;
        else if(size[v]*(n-size[v])>size[se[u]]*(n-size[se[u]]))
            se[u]=v;
    }
    size[u]++;tot[u]=d[u];//
    if(size[u]>1)
        d[u]+=size[fi[u]]*(n-size[fi[u]]);
}
void dfs2(int u,int fa)
{
    ans[u]=tot[u]+size[fi[u]]*(n-size[fi[u]])+pre[u];
    erg(i,u)
    {
        int v=edge[i].v;
        if(v==fa)continue;
        if(v==fi[u])//要移动到原重孩子~~
            pre[v]=ans[u]-d[v]-size[v]*(n-size[v])+size[se[u]]*(n-size[se[u]]);
        else
            pre[v]=ans[u]-d[v];    //完成外界信息的传递
        if(size[v]*(n-size[v])>size[fi[v]]*(n-size[fi[v]]))//修正以v为根的树的结构
            se[v]=fi[v],fi[v]=++m,size[m]=n-size[v];    //额外加点
        else if(size[v]*(n-size[v])>size[se[v]]*(n-size[se[v]]))
            se[v]=++m,size[m]=n-size[v];
        dfs2(v,u);
    }
}
int main()
{
    init();
    read(n);
    for1(i,n-1)
    {
        int a,b;
        read(a,b);
        addedge(a,b);
        addedge(b,a);
    }
    dfs1(1,0);
    m=n;
    dfs2(1,0);
    ll sum=0;
    for1(i,n)sum+=size[i]*(n-size[i]);
    for1(i,n)outln(sum-ans[i]);
}




你可能感兴趣的:(hihocoder)