ARC169D 做题记录

link

假定 $a_{1\sim n}$ 不对 $n$ 取模，设最终状态为 $b_{1\sim n}$，令 $S=\sum _{i=1}^n(b_i-a_i)$，应满足以下条件：

• $b_{i}modn$ 两两不同

• $m|S$

• $\underset{i=1}{\overset{n}{max}}(b_i-a_i)$

先对 $a$ 排序，那么可以发现最优情况下 $b$ 也是有序的。

• 再一个结论：$(b_1,b_2,\cdots ,b_n)=(x,x+1,\cdots ,x+n-1)$

考虑调整法，如果 $b_1+n\le b_n$，令 $b_1^{\prime }=b_n-n,b_n^{\prime }=b_1+n$ 再排序，依然满足条件。

那么我们需要令 $x$ 最小，枚举即可。

• 启示：不取模避免环的讨论，转化为其他条件。

点击查看代码

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define mkp make_pair
#define pir pair <ll, ll>
#define pb emplace_back
#define i128 __int128
void rd(ll &x) {
	char ch;
	while(!isdigit(ch = getchar())) ;
	x = ch - '0';
	while(isdigit(ch = getchar()))
		x = (x << 1) + (x << 3) + ch - '0';
}
using namespace std;
const ll maxn = 5e5 + 10, inf = 1e18;
ll n, m, a[maxn], w, ret, sum;
multiset <ll> st;
int main() {
	scanf("%lld%lld", &n, &m); w = n * (n - 1) / 2;
	for(ll i = 1; i <= n; i++)
		scanf("%lld", a + i), w -= a[i];
	w = (w % n + n) % n; ll stp = -1;
	for(ll i = 0; i < n; i++)
		if(i * m % n == w) { stp = i; break; }
	if(stp == -1) { puts("-1"); return 0; }
	sort(a + 1, a + 1 + n); ll d = 0;
	for(ll i = 1; i <= n; i++)
		d = max(d, a[i] - i + 1);
	for(ll i = 1; i <= n; i++) {
		sum += i - 1 + d - a[i];
		ret = max(ret, i - 1 + d - a[i]);
	}
	while(sum % m)
		sum += n, ++ret;
	ll g = m / __gcd(n, m);
	while(sum < ret * m)
		sum += n * g, ret += g;
	printf("%lld", sum / m);
	return 0;
}