每次判断奇偶性,如果偶数,就增加一个节点k-1,val为1,如果是奇数,就增加两个节点,一个为2个1节点,一个节点数为(k-1)/2,然后递归就好,就是把一个整体分成无数小状态做就好了。
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<stack>
#include<vector>
using namespace std;
#define N 100010
int fa[N],val[N],cnt=1;
void dfs(int k,int s)
{
if(k==1)
{
val[s]=1;
return;
}
if(k%2)
{
int s1=++cnt,s2=++cnt;
dfs(2,s1);
dfs((k-1)/2,s2);
fa[s1]=fa[s2]=s;
val[s]=val[s1]+val[s2];
}
else
{
int s1=++cnt;
dfs(k-1,s1);
fa[s1]=s;
val[s]=val[s1];
}
}
int main()
{
int k;
scanf("%d",&k);
if(k==1)
{
printf("2\n");
printf("1\n");
printf("1 2");
}
else
{
dfs(k,1);
printf("%d\n",cnt);
for(int i=2;i<=cnt;i++)
{
printf("%d ",fa[i]);
}
cout<<endl;
for(int i=1;i<=cnt;i++)
{
printf("%d ",val[i]);
}
}
cout<<endl;
return 0;
}