AtCoder Beginner Contest 353 C - Sigma Problem

链接：C - Sigma Problem

来源：AtCoder

题目描述：

题目解析：

这个问题要求我们计算一个序列中所有可能的两个不同元素的组合的函数f(x, y)的和。函数f(x, y)定义为(x+y)除以10^8的余数。我们需要计算的是所有这样的组合的余数之和。

首先，让我们理解一下函数f(x, y)。这个函数实际上是在计算两个数相加后的结果对10^8取余数。这意味着我们不需要关心两个数相加后的确切值，只需要知道它们相加后对10^8取余的结果。

接下来，我们需要考虑如何高效地计算这个问题的答案。由于N的范围很大（高达3×10^5），直接计算所有组合的f(x, y)将会非常慢，因为它的时间复杂度会是O(N^2)。

下面是比赛期间写的tle代码：

#include <bits/stdc++.h>
using namespace std;

int f(int x, int y) {
	return (x + y) % 100000000;
}

int main() {
	int N;
	cin >> N;
	vector<int> A(N);
	for (int i = 0; i < N; ++i) {
		cin >> A[i];
	}
	
	long long sum = 0;
	for (int i = 0; i < N - 1; ++i) {
		for (int j = i + 1; j < N; ++j) {
			sum += f(A[i], A[j]);
		}
	}
	
	cout << sum << endl;
	
	return 0;
}

结果显而易见。

优化分析：

比赛结束后寻找减少时间复杂度的办法，发现可以先对数组A进行排序，这一步是为了后续能够使用lower_bound进行二分查找，从而减少时间复杂度。lower_bound可以在有序数组中用于查找第一个不小于特定值的元素。在这个问题中，它用于快速计算有多少个元素与当前元素A[i]组合的和会超过100000000。我们可以计算一下优化完成后的时间复杂度，代码的时间复杂度主要来自于排序和遍历数组，分别是O(NlogN)和O(N)，因此总的时间复杂度是O(NlogN)。

下面为具体的AC代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

ll N;

int main() {
	cin >> N;
	vector<ll> A(N);
	ll sum = 0;
	for (int i = 0; i < N; i++) {
		cin >> A[i];
		sum += A[i];
	}
	
	sort(A.begin(), A.end());
	ll count = 0;
	
	for (int i = 0; i < N-1; i++){
		if(A[i] < 50000000) {
			if(A[i] + A[N - 1] >= 100000000) 
				count += A.end() - lower_bound(A.begin(), A.end(), 100000000LL - A[i]);
		}    
		else count += N - 1 - i;
	}
	
	ll ans = sum * (N - 1) - 100000000LL * count;
	
	
	
	cout << ans << "\n";
}

如有错误，望大佬们指正。