题目链接
题意: 要求构造一个有根树,这棵树上每个节点都有一个(权值)花费。树的所有叶子节点是怪物,在一个父亲节点上建造防御塔可以防御它这棵子数的所有怪物。给定一个方案数
K
K
K(
1
≤
K
≤
1
e
9
1\leq K\leq 1e9
1≤K≤1e9),要求构造出任意一个有根树,并且安排每个节点的权值(花费),使得能够防御住所有怪物、且总花费最小的不同建塔方案数恰好等于
K
K
K。
(当且仅当至少存在一个节点,一个方案中在这个节点建立了防御塔,而另一个方案没有在这个节点建立防御塔时,这两个方案被视为不同)
思路:
因为
K
K
K非常大,但总节点数量不能超过
1
e
5
1e5
1e5,所以考虑用
l
o
g
log
log2的结构来构造,即二叉树。
对于一棵二叉树,我们让X是根节点, L、R代表两个子树 ,并且,对于这棵树的每一层,我们让最底层节点的权值都为20,倒数第二层节点的权值都为21,… ,第二层节点的权值都为2d-2,根节点的权值为2d-1(d是这棵树的深度),因为这样可以满足根节点权值=两个子节点权值相加,所以我们就既可以选择两个子节点建立防御塔,也可以只选择根节点建立防御塔,花费相同。简略图如下:
X→R
↓
L
设这整棵树的方案数为
x
x
x,L子树的方案数为
a
a
a,R子树的方案数为
b
b
b,
那么可以得到这个关系式
x
=
a
∗
b
+
1
x=a*b+1
x=a∗b+1
(因为我们可以选择L子树中任意一种方案,然后选择R子树中任意一种方案,自由组合;也可以只选择根节点)
所以,如果可以让这个
b
b
b 固定为2,就可以把
K
K
K每次除以2,这样就可以实现用
l
o
g
log
log2
K
K
K的节点数量构造出一棵方案数恰好等于
K
K
K的有根树。
因此:
① 如果当前根节点的方案数
k
k
k是奇数,我们可以把它分为方案数为
(
k
−
1
)
/
2
(k-1)/2
(k−1)/2 和
2
2
2 的两棵子树,因为
(
k
−
1
)
/
2
∗
2
+
1
=
k
(k-1)/2*2+1 = k
(k−1)/2∗2+1=k。
② 如果当前根节点的方案数
k
k
k是偶数,我们可以把它分为方案数为
(
k
−
1
)
(k-1)
(k−1)和
1
1
1 的两颗子树,因为
(
k
−
1
)
∗
1
+
1
=
k
(k-1)*1+1=k
(k−1)∗1+1=k,从而把
k
k
k转换成①中奇数的情况。
注意:①和②两种方法,都是基于根节点权值=两个子节点权值相加 / 2这个条件。
这样就能实现用
l
o
g
log
log2
K
K
K的节点数量构造出一棵方案数恰好等于
K
K
K的有根树了。
但是要考虑: 节点权值最大为 1 e 9 1e9 1e9,一直这样分下去,树的深度使得根节点的权值2d-1可能大于 1 e 9 1e9 1e9,所以我们开始让根节点的权值等于 ≤ 1 e 9 \leq 1e9 ≤1e9的最大2的幂次方,每往下一层就除以2,直至当前节点权值为1时,就不能继续以①和②的方法往下分了,因为如果继续整除2往下分,将不满足根节点权值=两个子节点权值相加的这个条件。所以这个时候我们考虑可以用链式结构继续往下增加剩余的方案数(这个数不会太大),即X→Y→Z→…。在链上让父节点权值=子节点权值,因为这样的链式的树既可以满足根节点权值=子节点权值和,又可以每次增加一个方案。
另外,还需要注意特判 K = 1 K=1 K=1的情况,因为输出要求树的节点数 ≥ 2 \geq 2 ≥2
//AC代码
#include<bits/stdc++.h>
using namespace std;
const int Start=536870912,N=1e5+7;//Start=2^29
int k,n,fa[N],c[N];//n是节点数量,fa记录父亲节点,c记录节点权值
void solve(int k,int father,int value){//以递归的方式处理
int now=++n;
fa[now]=father;
c[now]=value;
if(k==1)return;
if(value==1){//value没法再除2, 后续只能以链的形式增加方案数
solve(k-1,now,1);
return;
}
if(k%2==1){
solve((k-1)/2,now,value/2);
solve(2,now,value/2);
}else{
solve(k-1,now,value/2);
solve(1,now,value/2);
}
}
int main(){
cin>>k;
if(k==1){//特判k=1的情况,因为输出要求树的节点数>=2
cout<<2<<"\n";
cout<<1<<"\n";
cout<<"2 1";
return 0;
}
solve(k,0,Start);
cout<<n<<"\n";
for(int i=2;i<=n;i++)printf("%d%c",fa[i]," \n"[i==n]);
for(int i=1;i<=n;i++)printf("%d ",c[i]);
}