题意:有n个人坐成一桌,定义一个人的沮丧程度为他喜欢的菜距离他的最小值,请随意旋转桌子,最小化所有人的沮丧程度。
思路:枚举我们沿顺时针方向旋转的次数,对于单个物体我们可以发现是一个分区间一次函数,那么我们只需要将所有的物体的函数求出来,累加即可,用差分就可以得到k和b,然后求解。
/*keep on going and never give up*/
#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
#define int long long
typedef pair<int, int> pii;
#define lowbit(x) x&(-x)
#define endl '\n'
#define wk is zqx ta die
int a[200005];
int pos[200005];
int k[200005];
int b[200005];
int n;
void add(int l, int r, int kk, int bb) {
if (l > r) {
return ;
}
k[l] += kk;
k[r + 1] -= kk;
b[l] += bb;
b[r + 1] -= bb;
}
void add(int v) {
if (v < n / 2) {
add(0, v, -1, v);
add(v + 1, v + n / 2, 1, -v);
add(v + n / 2 + 1, n - 1, -1, n + v);
} else {
v = n - v;
add(0, n / 2 - v, 1, v);
add(n / 2 - v + 1, n - v, -1, n - v);
add(n - v + 1, n - 1, 1, v - n);
}
}
signed main() {
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
pos[a[i]] = i;
}
int ans = 1e18;
for (int i = 0; i < n; i++) {
int v = (pos[i] - i + n) % n;
add(v);
}
for (int i = 1; i < n; i++) {
k[i] += k[i - 1];
b[i] += b[i - 1];
}
for (int i = 0; i < n; i++) {
ans = min(ans, k[i] * i + b[i]);
}
cout << ans << endl;
return 0;
}