1894C - Anonymous Informant - CodeForces Solution


constructive algorithms graphs implementation

Please click on ads to support us..

C++ Code:

#include <iostream>
#include <unordered_set>
using namespace std;

int n, k;
int b[200010];
bool vis[200010];

void solve() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> b[i];
    }
    for (int i = 1; i <= n; i++) {
        vis[i] = false;
    }

    unordered_set<int> s;
    int cur = n;
    bool ok = true;
    for (int i = 1; i <= k; i++) {
        if (b[cur] <= n) {
            if (vis[cur]) {
                break;
            } else {
                vis[cur] = true;
            }
            cur = ((cur - b[cur]) % n + n) % n;
            if (cur == 0) { cur = n; }
        } else {
            ok = false;
            break;
        }
    }

    cout << (ok ? "Yes\n" : "No\n");
}

int main() {
#ifdef ONLINE_JUDGE
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
#endif

    int t;
    cin >> t;
    for (int i = 1; i <= t; i++) {
        solve();
    }
}


Comments

Submit
0 Comments
More Questions

1302. Deepest Leaves Sum
1209. Remove All Adjacent Duplicates in String II
994. Rotting Oranges
983. Minimum Cost For Tickets
973. K Closest Points to Origin
969. Pancake Sorting
967. Numbers With Same Consecutive Differences
957. Prison Cells After N Days
946. Validate Stack Sequences
921. Minimum Add to Make Parentheses Valid
881. Boats to Save People
497. Random Point in Non-overlapping Rectangles
528. Random Pick with Weight
470. Implement Rand10() Using Rand7()
866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST