BZOJ 1429 方程的解

最新推荐文章于 2018-10-29 16:02:00 发布

skywalkert

最新推荐文章于 2018-10-29 16:02:00 发布

阅读量1.1k

点赞数

分类专栏： BZOJ 文章标签：数论

本文链接：https://blog.csdn.net/skywalkert/article/details/48217943

版权

BZOJ 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1429

题意：
设 $F(x)=\sum_{i=1}^{x}{i}$ ，求不定方程 $F (x) + F (y) + F (z) + F (w) = N$ 的自然数解个数。
$0\le N\le10^{12}$ 。

题解：
设 $x^2+y^2+z^2+w^2=N$ 的自然数解个数为 $S (n)$ ，N的约数之和为 $d (N)$ ，由雅可比的四平方和定理Jacobi’s four-square theorem可知
$S(2^km)=\begin{cases} & 8d(m) \text{ if } k=0 \\ & 24d(m) \text{ if } k>0\end{cases}(m\;is\;odd)$
而本题所求即为 $2x+1)^2+(2y+1)^2+(2z+1)^2+(2w+1)^2=4(2N+1)$ 的自然数解个数，这等价于 $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})$ 算一下即可。

代码：

#include <cstdio>
typedef long long LL;
LL n, ans;
int main()
{
	scanf("%lld", &n);
	n = (n << 1) + 1;
	ans = n + 1;
	for(LL i = 3; i <= n / i; ++i)
		if(n % i == 0)
		{
			ans += i;
			if(i != n / i)
				ans += n / i;
		}
	printf("%lld\n", ans);
	return 0;
}