题目分析
首先,x应当取序列a中某个元素的值。
那么我们就枚举所有x可能的值,此时我们要寻找最优的y。
记序列a中与x相邻的元素构成序列b。
如果一个数与x在左右两边都相邻,则在b中出现两次。
比如,对于a = {2, 2, 4, 1, 2, 1, 2, 3, 1},取x = 2,则b = {4, 1, 1, 1, 3}。
此时对于确定的y的取值,代价为
只有左边与y相关,因此常识最小化左侧式子的值。
这其实是一个经典问题,答案是取y为b的中位数。
那么算法其实很简单了,枚举所有x,求出序列b,对其排序并求出中位数。
由于所有b序列的长度之和不超过2n,因此算法复杂度为O(n log n)。
代码
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
int n, m, a[maxn];
ll ans, sum, bef, aft;
vector<int> b[maxn];
int main() {
scanf("%d%d", &n, &m);
for(int i = 1; i <= m; ++i) scanf("%d", a+i);
for(int i = 1; i <= m; ++i) {
if(i>1 && a[i-1]!=a[i])
b[a[i-1]].push_back(a[i]);
if(i<m && a[i+1]!=a[i])
b[a[i+1]].push_back(a[i]);
}
for(int i = 1; i <= n; ++i) {
if(!b[i].size()) continue;
sort(b[i].begin(), b[i].end());
int y = b[i][b[i].size()>>1];
bef = 0LL, aft = 0LL;
for(int j = 0; j < b[i].size(); ++j) {
bef += abs(i - b[i][j]);
aft += abs(y - b[i][j]);
}
ans = max(ans, bef-aft);
sum += bef;
}
printf("%d\n", sum/2-ans);
return 0;
}