[2023四校联考3]flandre

[2023四校联考3]flandre

题意

初始时，\(b_i=a_i\)。

对于 \(a_i<a_j\)，\(b_j \leftarrow b_j + k\)。

对于 \(a_i=a_j\)，\(b_j\) 不变。

对于 \(a_i>a_j\)，\(b_j \leftarrow b_j - k\)。

确定一种 \(a_i\) 的顺序，使得 \(b\) 的和最大。

思路

对于 \(a_i\) 全部大于 \(0\) 的情况，从小到大排序是最优的。

对于 \(a_i<0\) 的情况，先从小到大排序，再去掉没用的数。

没用的数指它造成的贡献是负数，即 \(a_i+\sum_{j=i+1}^{n} [a_i<a_j] \times k < 0\)。

从小到大去掉没用的数。

注意本题卡常，不要使用STL。

代码

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

void read(int& x) {
	x = 0; int f = 1;
	char ch = getchar();
	while (ch < '0' || ch > '9') {
		if (ch == '-') f = -f;
		ch = getchar();	
	}
	while (ch >= '0' && ch <= '9') {
		x = x * 10 + ch - '0';
		ch = getchar();
	}
	x *= f;
}

void write(int x) {
	if (x < 0) {
		x = -x;
		putchar('-');
	}
	if (x > 9) write(x / 10);
	putchar(x % 10 + '0');
}

const int N = 2e6 + 5;

int n, k, a[N], m, S[N];
int s[N];
pair <int, int> b[N];
vector <int> c; 

signed main() {
	freopen("flandre.in", "r", stdin);
	freopen("flandre.out", "w", stdout);
	read(n), read(k);
	for (int i = 1; i <= n; i ++) {
		read(a[i]);
		s[a[i]] ++;	
		b[i] = {a[i], i};
	}
	int sum = 0;
	for (int i = -1e6; i <= 1e6; i ++) {
		sum += s[i];
		if ((n - sum) * k + i >= 0) {
			S[i + (int)1e6] = 1;
			m += s[i];
		}
	}
	sort(b + 1, b + n + 1);
	for (int i = 1; i <= n; i ++) {
		if (S[b[i].first + (int)1e6]) {
			c.push_back(b[i].second);
		}
	}
	int add = 0, ans = 0;
	for (int i = 0; i < m; i ++) {
		ans += a[c[i]] + add;
		if (i != m - 1 && a[c[i]] != a[c[i + 1]]) {
			add += s[a[c[i]]] * k;
		}
	}
	write(ans); putchar(' '); 
	write(m); putchar('\n');
	for (int i = 0; i < m; i ++) {
		write(c[i]); putchar(' ');
	} 
	return 0;
}