Codeforces 433C Ryouko's Memory Note

题目链接：Codeforces 433C Ryouko's Memory Note

看了CLJ的代码，Orz~

使用vector记录一个数两边的数字，然后把这些数字排序，选中间的那个当做修改后的值，有两个地方需要注意，一个是push的时候如果有几个连续相同的数，这些数是不能放进对应的vector中的，第二个是下边枚举时如果用m会超时，所以记录一下页数的最大值，用页数枚举。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

const int MAX_N = 100000 + 100;

vector <int> V[MAX_N];
int arr[MAX_N];

int main()
{
    int n, m;
    long long temp = 0, _min, _beg = 0;
    int _max = 0;
    scanf("%d%d", &n, &m);
    for(int i = 0; i < m; i++)
    {
        scanf("%d", &arr[i]);
        _max = max(arr[i], _max);
        if(i != 0 && arr[i] != arr[i - 1])
        {
            V[arr[i - 1]].push_back(arr[i]);
            V[arr[i]].push_back(arr[i - 1]);
            _beg += abs(arr[i] - arr[i - 1]);
        }
    }
    _min = temp = _beg;
    int rev, _size;
    vector <int> k;
    for(int i = 1; i <= _max; i++)
    {
        temp = _beg;
        _size = V[i].size();
        if(_size == 0)
            continue;
        k = V[i];
        sort(k.begin(), k.end());
        rev = k[_size / 2];
        for(int j = 0; j < _size; j++)
            temp += (abs(rev - k[j]) - abs(i - k[j]));
        _min = min(temp, _min);
    }
    cout << _min << endl;
    return 0;
}