/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 532.0 KiB
#2 Accepted 2ms 324.0 KiB
#3 Accepted 2ms 344.0 KiB
#4 Accepted 269ms 35.945 MiB
#5 Accepted 200ms 35.336 MiB
#6 Accepted 270ms 36.086 MiB
#7 Accepted 268ms 36.09 MiB
#8 Accepted 276ms 39.504 MiB
#9 Accepted 279ms 52.344 MiB
#10 Accepted 199ms 35.168 MiB
#11 Accepted 197ms 35.086 MiB
#12 Accepted 198ms 35.039 MiB
#13 Accepted 199ms 35.016 MiB
#14 Accepted 196ms 34.75 MiB
#15 Accepted 198ms 34.793 MiB
#16 Accepted 203ms 34.887 MiB
#17 Accepted 205ms 34.957 MiB
#18 Accepted 200ms 34.918 MiB
#19 Accepted 206ms 34.695 MiB
#20 Accepted 231ms 34.848 MiB

Code

#include <bits/stdc++.h>
#define endl '\n'
#define F first
#define S second
using namespace std;
#define ll long long
#define pb push_back
#define mod 1000000007
#define int long long
#define all(x)      x.begin(),x.end()
#define allr(x)     x.rbegin(),x.rend()
#define CheckBit(x,k)   (x & (1LL << k))
#define SetBit(x,k)     (x |= (1LL << k))
#define ClearBit(x,k)   (x &= ~(1LL << k))
#define LSB(mask)       __builtin_ctzll(mask)
#define MSB(mask)       63-__builtin_clzll(mask) 
#define print(x)    cout << #x << " : " << x << endl
#define error1(x)   cerr << #x << " = " << (x) <<endl
#define coutall(v)  for(auto &it: v) cout<<it<<' '; cout<<endl
#define Abid_52     ios::sync_with_stdio(false);cin.tie(0),cin.tie(0)
#define error2(a,b) cerr<<"( "<<#a<<" , "<<#b<<" ) = ( "<<(a)<<" , "<<(b)<<" )\n"
#define UNIQUE(x)   sort(all(x)), x.erase(unique(all(x)), x.end()), x.shrink_to_fit()
template <typename T, typename U> T ceil(T x, U y) {return (x > 0 ? (x + y - 1) / y : x / y);}
template <typename T, typename U> T floor(T x, U y) {return (x > 0 ? x / y : (x - y + 1) / y);}

const int N = 2e5 + 10;

void solve()
{
    int n, m;
    cin >> n;
    vector<int> v(n+1);
    map<int,vector<int>>mp;
    map<int,int>On;
    for (int i = 1; i <= n; i++)
    {
        cin >> v[i];
        int on = v[i]+i;
        mp[on].push_back(i);
        On[i]=on;
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        if(mp[i].size()==0)
        {
            ans++;
        }
    }
    for(int i=1;i<=n;i++)
    {
        int next=On[i];
        if(next>n || i==n)
        {
            cout<<ans<<" ";
        }
        else
        {
            if(mp[next].size()==1 && *mp[next].begin()==i)
            {
                cout<<ans+1<<" ";
            } 
            else
            {
                cout<<ans<<" ";
            }    
        }
    }
    cout<<endl;
}

int32_t main()
{
    Abid_52;
    int t = 1;
    cin >> t;
    for (int tc = 1; tc <= t; ++tc)
    {
        // cout << "Case " << tc << ": ";
        solve();
    }
}

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:02:52
Judged At
2024-11-11 03:22:30
Judged By
Score
100
Total Time
279ms
Peak Memory
52.344 MiB