0819 T1 笔记

原题链接

题目分析

首先，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;
}