func

题目大意

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]);
}