BZOJ 1429 方程的解

题目链接

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

题解

题目相当于求
$$x^2+x+y^2+y+z^2+z+w^2+w=2N$$
即
$$(2x+1{)}^2+(2y+1{)}^2+(2z+1{)}^2+(2w+1{)}^2=4(2N+1)\text{\hspace{1em}(*)}$$
设$S(i)$表示$i$能拆成多少个由4个数的平方组成的和，$d(i)$表示$i$的约数和，由雅可比的四方定理（证明：arXiv:math/9206203），得到：
$$S(n)=S(2^{k}m)=\{\begin{array}{ll}8d(m)& k=0\\ 24d(m)& k\ge 1\end{array}$$
其中$n=2^{k}m$，$m$为奇数。

而$(\ast )$式的解的个数可以表示成$x^2+y^2+z^2+w^2=4(2N+1)$的解的个数减去$(2x{)}^2+(2y{)}^2+(2z{)}^2+(2w{)}^2=4(2N+1)$的解的个数并除去奇偶性和正负号的影响，即
$$\frac{S(4(2N+1))-S(2N+1)}{16}=d(2N+1)$$
$O(\sqrt{n})$求$d(2N+1)$即可。

代码

#include <cstdio>

long long n,ans;

int main()
{
  scanf("%lld",&n);
  n=n*2+1;
  for(long long i=1; i*i<=n; ++i)
    {
      if(n%i==0)
        {
          ans+=i;
          if(i*i<n)
            {
              ans+=n/i;
            }
        }
    }
  printf("%lld\n",ans);
  return 0;
}