/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 2ms 328.0 KiB
#2 Accepted 3ms 332.0 KiB
#3 Accepted 2ms 332.0 KiB
#4 Accepted 325ms 36.082 MiB
#5 Accepted 262ms 35.332 MiB
#6 Accepted 325ms 36.09 MiB
#7 Accepted 330ms 36.094 MiB
#8 Accepted 326ms 39.32 MiB
#9 Accepted 339ms 52.344 MiB
#10 Accepted 254ms 35.0 MiB
#11 Accepted 255ms 35.082 MiB
#12 Accepted 257ms 35.039 MiB
#13 Accepted 258ms 35.008 MiB
#14 Accepted 251ms 34.992 MiB
#15 Accepted 252ms 34.965 MiB
#16 Accepted 255ms 34.965 MiB
#17 Accepted 267ms 34.961 MiB
#18 Accepted 261ms 34.949 MiB
#19 Accepted 272ms 34.887 MiB
#20 Accepted 305ms 34.914 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-10-03 13:35:33
Judged By
Score
100
Total Time
339ms
Peak Memory
52.344 MiB