提出 #32716093

---

ソースコード 拡げる

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;

typedef long long int ll;
typedef unsigned long long int ull;
typedef modint1000000007 mint;
typedef modint998244353 mint2;

#define MOD 1000000007
#define MOD2 998244353
#define pi 3.1415926535
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rep2(i, s, n) for (int i = (int)(s); i < (int)(n); i++)
#define nqs(v) sort(v.begin(),v.end())
#define rev(v) reverse(v.begin(),v.end())

int main(void) {
    int N, K, Q;
    cin >> N >> K >> Q;
    vector<int> A(K);
    map<int,int> cnt;
    rep(i,K){
        cin >> A[i];
        cnt[A[i]] = 1;
    }
    
    rep(i,Q){
        int L;
        cin >> L;
        L--;
        if(A[L] == N){
            continue;
        } else {
            if(true){
                if(cnt[A[L]+1] == 0){
                    // ずらす
                    cnt[A[L]+1] = 1;
                    cnt[A[L]] = 0;
                    A[L]++;
                }
            }
        }
    }

    rep(i,K){
        cout << A[i] << " ";
    }
    return 0;
}

提出情報

ジャッジ結果