F. Almost All
题意:
第一行输入
n
(
1000
)
n(1000)
n(1000),表示树有
n
n
n个点;
接下来
n
−
1
n-1
n−1行,输入
u
,
v
u,v
u,v,表示树边;
现要求在树边上填任意正整数,要求树上任意两点距离在
[
1
,
⌊
2
n
2
9
⌋
]
[1,\lfloor{2n^2\over9}\rfloor]
[1,⌊92n2⌋]至少有一个值,输出任意一种方案。
题解:
发现
2
9
\frac29
92这个东西可以拆成
1
3
\frac 13
31和
2
3
\frac 23
32,然后就想把边分成两组
n
−
1
3
\frac{n-1}3
3n−1和
2
∗
(
n
−
1
)
3
\frac{2*(n-1)}{3}
32∗(n−1),使前面的子树集表示
[
1
,
n
−
1
3
]
[1,\frac{n-1}3]
[1,3n−1],后面子树集表示
[
(
n
−
1
3
+
1
)
∗
1
,
(
n
−
1
3
+
1
)
∗
2
∗
(
n
−
1
)
3
]
[(\frac{n-1}3+1)*1,(\frac{n-1}3+1)*\frac{2*(n-1)}3]
[(3n−1+1)∗1,(3n−1+1)∗32∗(n−1)],最后能表示的到
(
n
−
1
3
+
1
)
∗
(
2
∗
(
n
−
1
)
3
+
1
)
−
1
>
2
n
2
9
(\frac{n-1}3+1)*(\frac{2*(n-1)}3+1)-1>\frac{2n^2}9
(3n−1+1)∗(32∗(n−1)+1)−1>92n2;
至于如何实现?
1、首先分成两组,先找重心,然后以重心为根节点,把子树大小排序,从小取到大,大于等于
n
−
1
3
\frac{n-1}3
3n−1结束。
2、填数时从根节点dfs就好,从小到大填,儿子节点减父亲节点作为这条边的权值,具体看代码实现(这是一个用数学归纳法证明的定理,其实想想就知道了,靠感觉就可以)。
注意:
n
n
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;
}