JZOJ 3850. Fibonacci进制(fib)【思维】

---

题目：

传送门

---

题意：

将所有自然数用斐波那契进制表示出来，求前$n$位数字中有多少个$1$

---

分析：

我们观察$1,2,3,5,8,13\dots \dots$等数，我们会发现这些数在斐波那契进制的表示下只有最高位是$1$，其余数位都是$0$
之后我们将$n$进行分段，第一段，就是按照我们上面的将$n$不断分下去，直到没法再分为止；第二段，尽管第一段我们已经分到最后了(设此时的位数为$i$)，但显然还可能有多个位数为$i$的数，这样再去分；第三段，我们已经无法得到一个完整的数的斐波那契表示
然后逐一对上面的各个分段求解，累加在一起即是最终答案

---

代码：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#include<vector>
#define LL long long 
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
LL f[100],sum[100],len=1;
LL tf[100],ans=0;
void dfs(LL n)
{
	ans+=n;n--;
	if(n<=0) return;
	for(LL i=1;i<=len;i++)
	{
		if(n>=f[i-1]) n-=f[i-1];
		else
		{
			ans+=sum[i-1];
			if(n<=0) return;
			tf[i]=1;
			dfs(n);
			break;
		}
	}
	return;
}
int main()
{
	LL n=read();
	f[0]=f[1]=sum[1]=1;
	while(f[len]<1000000000000000ll) 
	{
		len++;
		f[len]=f[len-1]+f[len-2];
		sum[len]=f[len-1]+sum[len-1]+sum[len-2];
	}
	LL pos;
	for(LL i=1;i<=len;i++)
	{
		if(n>=f[i-1]*i) n-=f[i-1]*i;
		else
		{
			ans=sum[i-1];
			pos=i;
			break;
		}
	}
	tf[pos]=1;
	dfs(n/pos);
	tf[1]++;
	for(LL i=1;i<=len;i++) if(tf[i]&&tf[i+1]) tf[i+2]=1,tf[i]=tf[i+1]=0;
	for(LL i=n%pos,j=pos;i;i--,j--) ans+=tf[j];
	cout<<ans;
	return 0;
}