A. Hide and Seek-CSDN博客

本文链接：https://blog.csdn.net/yingxueyuhanling/article/details/144633123

#include <iostream>
#include <vector>

using namespace std;

// 思路：假如 Bob 啥也没说，先求出所有情况为 n*3-2，再
// 将不符合条件的情况减去即可，求不符合条件的情况：
//    1、Bob 说了一个数 x，就不存在情况(x,x)了
//    2、Bob 先说 x，后说 y，就不存在情况(x,y)了
// 利用一个二维 vis 数组，序列从后往前遍历
//    vis[arr[i]][0]==true,说明情况(arr[i],arr[i])统计过了
//    vis[arr[i]-1][0]==true,说明情况(arr[i]-1，arr[i])统计过了
//    vis[arr[i]+1][0]==true,说明情况(arr[i]，arr[i]+1)统计过了
vector<int> arr(100005);
vector<vector<bool>> vis(100005, vector<bool>(3, false));


int main() {
    int n, k;
    cin >> n >> k;
    for (int i = 0; i < k; i++) {
        cin >> arr[i];
    }
    int ans = n * 3 - 2;
    for (int i = k - 1; i >= 0; i--) {
        if (!vis[arr[i]][0]) {
            vis[arr[i]][0] = true;
            ans--;
        }
        if (!vis[arr[i]][1] && vis[arr[i] - 1][0]) {
            vis[arr[i]][1] = true;
            ans--;
        }
        if (!vis[arr[i]][2] && vis[arr[i] + 1][0]) {
            vis[arr[i]][2] = true;
            ans--;
        }
    }
    cout << ans << endl;
    // 清空 vector
    arr.clear();
    vis.clear();
    return 0;
}