Codeforces Round 908 (Div. 2) C. Anonymous Informant (思维题)

题目链接

思路:

        考虑到 k 的范围很大, 肯定不能进行模拟。

        通过观察以及题目的条件, a[i] = i 时是定点, 并循环左移 i 次, 则左移后, 数组的最后一位即为该定点, 且通过这个定点可以知道此定点未左移时, 数组的第一个数是什么, 也就可以知道当时数组的最后一个数是什么, 即又找到前一个定点, 不断重复这个过程, 则所有的定点都可以找到。

        当最后一个数 > n时, 显然不是定点, 则输出No即可

        考虑到 k 的范围很大, 则找k次肯定会超时。观察发现找定点的操作在几次之后就会重复，则长为n的数组最多不超过 n 个顶点， 则找定点的操作为 min（n， k） 

代码：

#include<bits/stdc++.h>
using namespace std;
#define endl "\n"
typedef long long ll;
typedef pair<int, int> PII;
typedef pair<PII, int> PIII;
const int inf = 0x3f3f3f3f;
const ll infinf = 0x3f3f3f3f3f3f3f3f;

//const int N = 

void solve() {
    int n, k; cin >> n >> k; 
    vector<int> a(n + 10);
    
    for (int i = 1; i <= n; i ++) cin >> a[i];

    int cnt = min(n, k), last = n;
    while (cnt --) {
        if (a[last] > n) {
            cout << "No" << endl;
            return;
        }
        else {
            last -= a[last];        // 找前一个定点
            if (last <= 0) last += n;  // 循环左移, 则last可能为负
        }
    }
    cout << "Yes" << endl;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int t; cin >> t;
    while (t --) solve();

    return 0;
}