CF1305E Kuroni and the Score Distribution（构造）

Description

请你构造一个长为 $n$ 的序列，使得满足 $a_i + a_j = a_k$ 的三元组 $(i,j,k)$ 恰好有 $m$ 个。无解输出 $-1$。

$1 \leq n \leq 5000, 1 \leq m \leq 10^9$。

Solution

不难想到按 $1,2,3 \dots$ 的方法填，那么求一下以每个位置作为 $k$ 的三元组个数，可以发现 $a_1 + a_{k-1} = a_2+a_{k-2} = \dots = a_{k-1}+a_1 = a_k$。因为有重复的，所以贡献为 $\left \lfloor\frac{k - 1}{2} \right\rfloor$。一直填到 $m \leq \sum_i \left \lfloor\frac{k - 1}{2} \right\rfloor$。如果填不到那么无解。

可以发现让最后一个数 $+2$ 它的贡献为 $\left \lfloor\frac{k - 3}{2} \right\rfloor = \left \lfloor\frac{k - 1}{2} \right\rfloor - 1$。所以给最后一个数加上 $2 \times (- m + \sum_i \left \lfloor\frac{k - 1}{2} \right\rfloor)$。可能还没有填完，因为 $n$ 的规模小，所以从 $10^9$ 每次递减最后一个数加一来填。时间复杂度为 $O(n)$。

Code

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5, INF = 0x3f3f3f3f;
inline int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
int ans[N];
int main() {
	int n = read(), m = read();
	int sum = 0, i;
	for (i = 1; i <= n && sum < m; i++) sum += (i - 1) / 2, ans[i] = i;
	if (sum < m) {
		puts("-1"); return 0;
	}
	ans[i - 1] += (sum - m) * 2; sum = 1e9;
	for (int j = n; j >= i; j--) ans[j] = sum, sum -= ans[i - 1] + 1;
	for (int i = 1; i <= n; i++) printf("%d ", ans[i]);
	return 0;
}