BZOJ 5399 illustrious

题目链接

https://www.lydsy.com/JudgeOnline/problem.php?id=5399

题解

打表观察$f$序列，我们发现$f(i)$就是$i$这个数在$f$这个序列中出现的次数。

因此，如果我们暴力算出$f$的前$10^6$项的值，我们就可以求出$f$的前$3\times 10^9$项的值。

而$g$序列是$f$的前缀和，那么$g(n)$的含义就是$f(x)=n$的最大的$x$。

考虑$h(n)=h(g(f(n))-f(f(n)))+g(g(n))$中每一项的含义。

$f(f(n))$就是$f(n)$的出现次数。

$g(f(n))$就是序列中最大为$f(n)$的位置。

那么$g(f(n))-f(f(n))$就是序列中最大的等于$f(n)-1$的位置，即$g(f(n)-1)$。

因此$h(n)$就是$n$位置之前的每一段数的结尾位置的$g(g(n))$之和。

打表发现$g(g(n))=\sum_{i=1}^n i\times f(i)$，可以边算 h h <script type="math/tex" id="MathJax-Element-26">h</script>边算这个。

代码

#include <cstdio>

int read()
{
  int x=0,f=1;
  char ch=getchar();
  while((ch<'0')||(ch>'9'))
    {
      if(ch=='-')
        {
          f=-f;
        }
      ch=getchar();
    }
  while((ch>='0')&&(ch<='9'))
    {
      x=x*10+ch-'0';
      ch=getchar();
    }
  return x*f;
}

const int maxm=1000000;
const int mo=998244353;
const int half=(mo+1)>>1;

int f[maxm+10];

int main()
{
  f[1]=1;
  for(int i=2; i<=maxm; ++i)
    {
      f[i]=f[i-f[f[i-1]]]+1;
    }
  int t=read();
  while(t--)
    {
      int n=read(),tot=0,g=0,i=1,res=0;
      while(tot+f[i]<=n)
        {
          int tmp=(2ll*tot+f[i]+1)%mo*f[i]%mo*half%mo;
          g=(g+1ll*tmp*i)%mo;
          res=(res+g)%mo;
          tot+=f[i++];
        }
      if(tot<n)
        {
          int tmp=(1ll*tot+n+1)%mo*(n-tot)%mo*half%mo;
          g=(g+1ll*tmp*i)%mo;
          res=(res+g)%mo;
        }
      printf("%d\n",res);
    }
  return 0;
}