Atcoder ABC292 C - Four Variables题解

最新推荐文章于 2024-07-23 22:06:37 发布

BreaCham

最新推荐文章于 2024-07-23 22:06:37 发布

阅读量349

点赞数

分类专栏： Atcoder个人题解文章标签：算法 c++

本文链接：https://blog.csdn.net/BreaCham/article/details/129390093

版权

Atcoder个人题解专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述

You are given a positive integer N.
Find the number of quadruples of positive integers (A,B,C,D) such that AB+CD=N. Under the constraints of this problem, it can be proved that the answer is at most $9×10^{18}$ .

分析

要求AB+CD=N的组合数，如果暴力枚举将达到 $N^3$ 的复杂度。

将问题分解，求出满足 $A B = X, C D = Y, X + Y = N$ 的个数。那么只要确定X,A,C即可，此时复杂度降为了 $N^2$ 。继续简化：不妨设 $A\le B$ ,那么A只会取到 $\le \sqrt N$ 的值，复杂度为 $O(n\sqrt n)$ 。

本题关键：将问题拆分为两个小的求和问题的叠加，可以使复杂度从乘积的数量级降到求和的数量级。另外，很多数学问题枚举因数时只需枚举到根号之下即可。

代码

#include<bits/stdc++.h>

using namespace std;

int main() {
	int n;
	cin >> n;

	int x, y;
	long long ans = 0;
	for (int i = 0; i <= n; ++i) {
		x = i, y = n - i;
		
		long long cnt1 = 0, cnt2 = 0;
		for (int j = 1; j * j <= x; ++j) {
			if (x % j == 0) {
				++cnt1;
				if (j * j != x)
					++cnt1;
			}
		}

		for (int j = 1; j * j <= y; ++j) {
			if (y % j == 0) {
				++cnt2;
				if (j * j != y)
					++cnt2;
			}
		}
		ans += cnt1 * cnt2;
	}
	cout << ans;
	return 0;
}