16.1113 模拟考试T1

笔记
【问题描述】
给定一个长度为m的序列a，下标编号为1~m。序列的每个元素都是1~N的
整数。定义序列的代价为累加（1->m-1 abs(ai+1-ai)）
你现在可以选择两个数x和y，并将序列?中所有的x改成y。x可以与y相等。
请求出序列最小可能的代价。
【输入格式】
输入第一行包含两个整数n和m。第二行包含n个空格分隔的整数，代表序
列a。
【输出格式】
输出一行，包含一个整数，代表序列最小的代价。
【样例输入 1】
4 6
1 2 3 4 3 2
【样例输出 1】
3
【样例输入 2】
10 5
9 4 3 8 8
【样例输出 1】
6
【样例解释】
样例 1 中，最优策略为将 4 改成 3。样例 2 中，最优策略为将 9 改成 4。
【数据规模和约定】
对于30%的数据，n,m<=100
对于60%的数据，n,m ≤ 2000。
对于100%的数据，1 ≤ n,m ≤ 100,000。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <ios>
#include <vector>
using namespace std;
typedef long long ll;
const int N = (int)1e5;
int n, m, a[N+1];
vector<int> b[N+1];
int main(int argc, char *argv[]) {
freopen("note.in", "r", stdin);
freopen("note.out", "w", stdout);

ios :: sync_with_stdio(false);
cin>>n>>m;
for (int i=1;i<=m;++i) cin>>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]);
}

ll ans=0LL,sum=0LL;
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];
ll before=0LL,after=0LL;
for (int j=0;j<b[i].size();++j) {
before+=abs(i-b[i][j]);
after+=abs(y-b[i][j]);
}
ans=max(ans,before-after),sum+=before;
}
cout<<sum/2-ans<<endl;
fclose(stdin);
fclose(stdout);
return 0;
}

思路：动态数组记录下与每个数字相邻的所有数字，若是要修改这这个数字的话，肯定是从与它相邻的数字中间取中位数（先sort）这样对于答案的贡献肯定最小,before记录改之前的值，sum记录所有改之前的值，sum/2是因为相邻的两个数字肯定是互相相邻的，即每一个数字都被计算了两遍，ans记录对答案贡献较大的修改值，即遍历一遍检查修改1-N中的哪个数字，可以使答案更优，before-after为将i修改为

int y=b[i][b[i].size()>>1]前后差值，即修改i对答案的优化程度

数据：见网盘我叫雷锋~ 提取密码：gmcf