/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 532.0 KiB
#2 Accepted 2ms 396.0 KiB
#3 Accepted 1ms 532.0 KiB
#4 Accepted 31ms 3.062 MiB
#5 Accepted 34ms 3.82 MiB
#6 Accepted 30ms 3.07 MiB
#7 Accepted 32ms 3.18 MiB
#8 Accepted 38ms 3.754 MiB
#9 Accepted 43ms 4.062 MiB
#10 Accepted 35ms 3.961 MiB
#11 Accepted 35ms 3.828 MiB
#12 Accepted 35ms 3.816 MiB
#13 Accepted 36ms 3.848 MiB
#14 Accepted 35ms 3.816 MiB
#15 Accepted 35ms 3.816 MiB
#16 Accepted 35ms 3.812 MiB
#17 Accepted 35ms 3.859 MiB
#18 Accepted 36ms 3.816 MiB
#19 Accepted 36ms 3.848 MiB
#20 Accepted 37ms 3.816 MiB

Code

// I AM A MUSLIM

#include "bits/stdc++.h"

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")

#define fastIO std::ios::sync_with_stdio(0);std::cin.tie(0)
#define ll long long int
#define flush fflush(stdout)
#define bl printf("\n")
#define yn(a, b) printf("%s\n", a >= b ? "Yes":"No")
// #define int ll

using pii = std::pair<int,int>;

const int MOD = 1000000007;
// const int MOD = 998244353;
const int mxN = 500005, inf = INT_MAX;

int par[mxN], lnk[mxN];

signed main() {
    // fastIO;
    int testCases=1;
    scanf("%d",&testCases);
    // std::cin>>testCases;
    
    for (int TC = 1; TC <= testCases; TC++) {
        int N;
        scanf("%d",&N);
        int tot = 0;
        for (int i = 0; i < N; i++) par[i] = 0, lnk[i] = -1;
        
        int a[N]; for (auto &i : a) scanf("%d",&i);
        
        for (int i = 0; i < N; i++) {
            int at = i+a[i];
            if (at < N) lnk[i] = at, par[at]++;
            if (par[i]) continue;
            tot++;
        }
        // printf("%d\n", tot);
        
        for (int i = 0; i < N; i++) {
            int L = lnk[i];
            // printf("(%d %d)\n", i, L);
            int ans = tot;
            if (L != -1 && par[L] == 1) ans++;
            printf("%d ",ans);
        } bl;
        
    }
    
    return 0;
}

/*

*/

Information

Submit By
Type
Submission
Problem
P1066 Light switches
Contest
Brain Booster #4
Language
C++20 (G++ 13.2.0)
Submit At
2024-07-14 18:33:30
Judged At
2024-11-11 03:22:13
Judged By
Score
100
Total Time
43ms
Peak Memory
4.062 MiB