题目大意
f(1)=1,f(2i)=f(i),f(2i+1)=f(i)+f(i+1)。
求f(m)=n的所有m。
考虑
如何计算一个f(m),先拆二进制位。
用f0和f1表示后面的位做完用多少进位和不进位。
我们只考虑最后一位是1的也就是奇数,那么初始f0=f1=1。
考虑一位上是1因为f(2i+1)=f(i)+f(i+1),所以我们将它减半后可以选择+1。
那么当一位是1时,如果后面没有进位,它可以选择往前影响或不影响。如果有进位,就会继续进位,而这一位为0,f(2i)=f(i)。
所以此时f1+=f0。
当一位是0时类似有f0+=f1。
因为最后一位一定是1所以f1>=f0,而且f1就是f(m)。
发现这个过程像欧几里得,因为我们可以枚举f0(gcd(f0,f1)=1),然后模拟gcd的过程得到方案。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=1000000+10,mo=998244353;
int ans[maxn],two[maxn];
int i,j,k,l,t,n,m,top,now;
int gcd(int a,int b){
return b?gcd(b,a%b):a;
}
void likegcd(int f0,int f1){
if (f1==0) return;
if (f0<=f1){
int t=f1/f0;
now=((ll)now*two[t]%mo+two[t]-1)%mo;
likegcd(f0,f1%f0);
}
else{
int t=f0/f1;
if (f0%f1==0) t--;
now=(ll)now*two[t]%mo;
likegcd(f0-t*f1,f1);
}
}
int main(){
freopen("func.in","r",stdin);freopen("func.out","w",stdout);
scanf("%d",&n);
two[0]=1;
fo(i,1,n) two[i]=(ll)two[i-1]*2%mo;
fo(i,1,n)
if (gcd(i,n)==1){
now=0;
likegcd(i,n);
ans[++top]=now;
}
sort(ans+1,ans+top+1);
fo(i,1,top) printf("%d\n",ans[i]);
}