Codeforces Round #580 (Div. 2)F. Almost All(思维(数学分块)+树的重心)

F. Almost All

题意：

第一行输入$n(1000)$，表示树有$n$个点；
接下来$n-1$行，输入$u,v$，表示树边；
现要求在树边上填任意正整数，要求树上任意两点距离在$[1,\lfloor \frac{2n^2}{9}\rfloor ]$至少有一个值，输出任意一种方案。

题解：

发现$\frac{2}{9}$这个东西可以拆成$\frac{1}{3}$和$\frac{2}{3}$,然后就想把边分成两组$\frac{n-1}{3}$和$\frac{2\ast (n-1)}{3}$,使前面的子树集表示$[1,\frac{n-1}{3}]$,后面子树集表示$[(\frac{n-1}{3}+1)\ast 1,(\frac{n-1}{3}+1)\ast \frac{2\ast (n-1)}{3}]$,最后能表示的到$(\frac{n-1}{3}+1)\ast (\frac{2\ast (n-1)}{3}+1)-1>\frac{2n^2}{9}$；
至于如何实现？
1、首先分成两组，先找重心，然后以重心为根节点，把子树大小排序，从小取到大，大于等于$\frac{n-1}{3}$结束。
2、填数时从根节点dfs就好，从小到大填，儿子节点减父亲节点作为这条边的权值，具体看代码实现(这是一个用数学归纳法证明的定理，其实想想就知道了，靠感觉就可以)。
注意：$n$为1的情况要考虑。

代码：

#include<bits/stdc++.h>
using namespace std;
int n;
vector<int>g[1009];
int siz[1009],w[1009],mx=1009,rt;
void dfs1(int u,int fa){
    siz[u]=1;
    for(auto v:g[u])if(v!=fa){
        dfs1(v,u);
        siz[u]+=siz[v];
        w[u]=max(w[u],siz[v]);
    }
    w[u]=max(w[u],n-siz[u]);
    if(mx>w[u])mx=w[u],rt=u;
}
void dfs2(int u,int fa){
    siz[u]=1;
    for(auto v:g[u])if(v!=fa){dfs2(v,u);siz[u]+=siz[v];}
}
int tot,z[1009],base;
void dfs3(int u,int fa){
    tot+=base;z[u]=tot;
    cout<<u<<" "<<fa<<" "<<tot-z[fa]<<endl;
    for(auto v:g[u])if(v!=fa)dfs3(v,u);
}
#define P pair<int,int>
P a[1009];int na;
int main(){
   // freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    cin>>n;
    for(int i=1,u,v;i<n;i++)cin>>u>>v,g[u].push_back(v),g[v].push_back(u);
    dfs1(1,0);dfs2(rt,0);
//    cout<<rt<<endl;
    for(auto v:g[rt])a[++na]=P(siz[v],v);
    sort(a+1,a+na+1);
    int mid=0,sum=0;
    for(int i=1;i<=na;i++){
        sum+=a[i].first;
        if(sum>=(n-1)/3){mid=i;break;}
    }
  //  cout<<mid<<endl;
    base=1;
    for(int i=1;i<=mid;i++){
        dfs3(a[i].second,rt);
    }
    base=tot+1;tot=0;
    for(int i=mid+1;i<=na;i++){
        dfs3(a[i].second,rt);
    }
    return 0;
}