题目链接
思路:
考虑到 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;
}