AtCoder Beginner Contest 277 D Takahashi's Solitaire

Takahashi's Solitaire

双指针 && 尺取

先排个序，然后把数组扩展到 \(2 \times n\)，为了处理循环的情况

然后双指针跑一下，限定 \(r\) 扩展的条件为：当前数字等于前一个 或者 比前一个大 \(1\) 或者 尺取的第一个点

为了防止循环重复的问题，要限制一下尺取的长度不超过 \(n\)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <queue>
#include <functional>
#include <map>
#include <set>
#include <cmath>
#include <cstring>
#include <deque>
#include <stack>
#include <ctime>
#include <cstdlib>
using namespace std;
typedef long long ll;
#define pii pair<int, int>
const ll maxn = 2e5 + 10;
const ll inf = 1e17 + 10;

int main()
{
    int n, m;
    cin >> n >> m;
    vector<ll>a(n * 2);
    ll sum = 0;
    for(int i=0; i<n; i++) {cin >> a[i]; sum += a[i];}
    for(int i=0; i<n; i++) a[i + n] = a[i] + m;
    sort(a.begin(), a.end());
    int l = 0, r = 0;
    ll now = 0, ans = sum, cnt = 0;
    while(l < n * 2)
    {
        while(r < n * 2 && (l == r || a[r] == a[r - 1] || a[r] == a[r - 1] + 1) && cnt < n)
        {
            now += a[r] >= m ? a[r] - m : a[r];
            r++;
            cnt++;
        }
        ans = min(ans, sum - now);
        now -= a[l] >= m ? a[l] - m : a[l];
        l++;
        cnt--;
    }
    cout << ans << endl;
    return 0;
}