G. Game Design
来源:2019-2020 ICPC Asia Hong Kong Regional Contest
分析:
首先看到节点最多只有2E5个,所以暴力写出1E9个节点是不可以的。
再思考只有最小开销的策略才会被统计,所以我们可以利用这一点来将无用点设为一个极大的数来避免干扰。
根据这点我们使用如下方法:
+------+
|X/2+1?|
+------+
| |
+--++ ++--+
|X/2| | 1 |
+--++ ++--+
| |
+ ++--+
...... | 1 |
+---+
对于一个X,将它分两个子节点,左节点权重等于X/2,右结点为权重为1的两个节点连起来。
然后再讨论奇数偶数的情况:
- 奇数:当前节点权重设为左节点权重加1,则总方案数为左节点选法*2+1
- 偶数:当前节点权重设为一个极大的数字,则总方案数为左节点选法*2
根据如上规律递推建树即可。
因为题目特别提到节点数至少为2,所以特别处理下K=1的情况。
#include <bits/stdc++.h>
using namespace std;
const int MAXN=2e5+5;
int fa[MAXN];
int w[MAXN];
bool unshow[MAXN];
int cur=1;
int dfs(int x,int f){
int u=cur;
if(x<=0) return 0;
else if(x==1){
fa[u]=f;
w[u]=1;
cur++;
return w[u];
}
fa[cur]=f;
cur++;
w[cur]=1;
fa[cur]=u;
cur++;
w[cur]=1;
fa[cur]=cur-1;
cur++;
w[u]=dfs(x/2,u)+1;
if(x%2==0) unshow[u]=true;
return w[u];
}
int main(){
int n;
scanf("%d",&n);
if(n==1){
printf("3\n1 1\n100 1 1");
return 0;
}
dfs(n,0);
printf("%d\n",cur-1);
for(int i=2;i<cur;i++) printf("%d ",fa[i]);
printf("\n");
const int big=1e9-7;
for(int i=1;i<cur;i++){
if(unshow[i]) printf("%d ",big);
else printf("%d ",w[i]);
}
return 0;
}