結果

問題 No.1373 Directed Operations
ユーザー gunmanekogunmaneko
提出日時 2021-02-05 21:53:18
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 6,843 bytes
コンパイル時間 2,862 ms
コンパイル使用メモリ 185,532 KB
実行使用メモリ 7,140 KB
最終ジャッジ日時 2024-07-02 12:24:11
合計ジャッジ時間 4,485 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,816 KB
testcase_01 AC 3 ms
6,944 KB
testcase_02 AC 144 ms
7,136 KB
testcase_03 AC 22 ms
7,016 KB
testcase_04 AC 34 ms
7,008 KB
testcase_05 AC 3 ms
6,944 KB
testcase_06 AC 153 ms
7,140 KB
testcase_07 AC 7 ms
6,944 KB
testcase_08 AC 19 ms
6,940 KB
testcase_09 AC 37 ms
7,008 KB
testcase_10 AC 33 ms
7,012 KB
testcase_11 AC 11 ms
6,940 KB
testcase_12 AC 32 ms
7,008 KB
testcase_13 AC 7 ms
6,940 KB
testcase_14 AC 157 ms
7,012 KB
testcase_15 AC 135 ms
7,012 KB
testcase_16 AC 157 ms
7,016 KB
testcase_17 AC 113 ms
7,016 KB
testcase_18 AC 66 ms
6,940 KB
testcase_19 AC 65 ms
6,940 KB
testcase_20 AC 89 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
//#include <atcoder/all>
//using namespace atcoder;
using namespace std;
const int MAX = 700000;
const int MOD = 1000000007;
const long long INF = 1LL << 60;
long long  fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++) {
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}

// 二項係数計算
long long COM(int n, int k) {
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}
/*第二引数で第一引数を割ったときの切り上げの計算*/
long long int maxtime(long long int x, long long int y) {
    return(x + y - 1) / y;

}
/*最大公約数*/
long long int lcm(long long int number1, long long int number2) {
    long long int m = number1;
    long long int n = number2;

    if (number2 > number1) {
        m = number2;
        n = number1;
    }
    long long int s = -1;
    while (s != 0) {
        s = m % n;
        m = n;
        n = s;
    }
    return m;
}
/*最大公倍数*/
long long int gcd(long long int number1, long long int number2) {
    long long int m = number1;
    long long int n = number2;
    return m / lcm(m, n) * n;
}
/*逆元計算*/
long long int  modinv(long long a, long long m) {
    long long int b = m, u = 1, v = 0;
    while (b) {
        long long int t = a / b;
        a -= t * b; swap(a, b);
        u -= t * v; swap(u, v);
    }
    u %= m;
    if (u < 0) u += m;
    return u;
}
long long mod(long long val, long long m) {
  long long res = val % m;
  if (res < 0) res += m;
  return res;
}

//セグメントツリー(あってるかはわからんから自己責任で)
const int MAX_N = 10000000;
//セグメント木を持つグローバル配列
int n,dat[2*MAX_N - 1];

//初期化
void init(int n_){
    //簡単のため，要素数を２のべき乗に
    n = 1;
    while(n < n_) {
        n *= 2;
    }
    //すべての値をINT_MAXに
    for(int i = 0; i < 2 * n - 1; i++){
        //dat[i] = INT_MAX;
        dat[i] = 0;
    }
}
// k番目の値(0-indexed)をaに変更
void update(int k,int a){
    //葉の節点
    k += n - 1;
    dat[k] = a;
    //登りながら更新
    while(k > 0){
        k = (k - 1) / 2;
        dat[k] = max(dat[k * 2 + 1], dat[k * 2 + 2]);
    }
}
//　[a, b)の最小値を求める
//　後ろのほうの引数は，計算の簡単のための引数
//kは節点の番号, l, rはその節点が[l, r)に対応づいていることを表す。
//　したがって，外からはquery(a, b, 0, 0, n)として呼ぶ．
int query(int a, int b, int k = 0, int l = 0, int r = n){
    // [a, b)と[l, r)が交差しなければ，INT_MAX
    if (r <= a || b <= l){ 
        return 0;
    }
    // [a, b)が[l, r)を完全に含んでいれば，この節点の値
    if (a <= l && r <= b){ 
        return dat[k];
    }else{
        //そうでなければ，2つの子の最小値
        int vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
        int vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
        return max(vl,vr);
    }
}

//Union-Find　アッカーマン関数よくわからない
int par[MAX_N];     //親
int ranks[MAX_N];    //木の深さ

//n要素で初期化
void initunion(int n){
    for(int i = 0;i < n;i++){
        par[i] = i;
        ranks[i] = 0;
    }
}

//木の根を求める
int find(int x){
    if (par[x] == x){
        return x;
    }else{
        return par[x] = find(par[x]);
    }
}
//xとyの属する集合を併合
void unite(int x,int y){
    x = find(x);
    y = find(y);
    if(x == y){
        return;
    }
    if(ranks[x] < ranks[y]){
        par[x] = y;
    }else{
        par[y] = x;
        if(ranks[x] == ranks[y]){
            ranks[x]++;
        }
    }
}
//xとyが同じ集合に属するか否か
bool same(int x,int y){
    return find(x) == find(y);
}

//めぐる式二部探索！サノバウィッチやったことない
// index が条件を満たすかどうか
vector<long long int >meguru;
// index が条件を満たすかどうか
bool isOK(int index, int key) {
    if (meguru[index] >= key) return true;
    else return false;
}
// 汎用的な二分探索のテンプレ
int binary_search(int key) {
    int left = -1; //「index = 0」が条件を満たすこともあるので、初期値は -1
    int right = (int)meguru.size(); // 「index = a.size()-1」が条件を満たさないこともあるので、初期値は a.size()
    /* どんな二分探索でもここの書き方を変えずにできる！ */
    while (right - left > 1) {
        int mid = left + (right - left) / 2;

        if (isOK(mid, key)) right = mid;
        else left = mid;
    }
    /* left は条件を満たさない最大の値、right は条件を満たす最小の値になっている */
    return right;
}
long long modpow(long long a, long long n, long long mod) {
    long long res = 1;
    while (n > 0) {
        if (n & 1) res = res * a % mod;
        a = a * a % mod;
        n >>= 1;
    }
    return res;
}

struct Edge {
    long long to;
    long long cost;
};
using Graph = vector<vector<Edge>>;
using P = pair<long, int>;
/* dijkstra(G,s,dis)
    入力：グラフ G, 開始点 s, 距離を格納する dis
    計算量：O(|E|log|V|)
    副作用：dis が書き換えられる
*/
void dijkstra(const Graph &G, int s, vector<long long> &dis) {
    int N = G.size();
    dis.resize(N, INF);
    priority_queue<P, vector<P>, greater<P>> pq;  // 「仮の最短距離, 頂点」が小さい順に並ぶ
    dis[s] = 0;
    pq.emplace(dis[s], s);
    while (!pq.empty()) {
        P p = pq.top();
        pq.pop();
        int v = p.second;
        if (dis[v] < p.first) {  // 最短距離で無ければ無視
            continue;
        }
        for (auto &e : G[v]) {
            if (dis[e.to] > dis[v] + e.cost) {  // 最短距離候補なら priority_queue に追加
                dis[e.to] = dis[v] + e.cost;
                pq.emplace(dis[e.to], e.to);
            }
        }
    }
}
int main(){
    long long int n;
    cin >> n;
    vector<pair<long long int ,long long int >>av;
    vector<long long int >bv;
    long long int dp[200000] = {};
    for(long long int i = 1;i < n;i++){
        long long int a;
        cin >> a;
        av.push_back(make_pair(a,i));
    }
    sort(av.begin(),av.end());
    for(long long int i = 0;i < av.size();i++){
        if(av[i].first > i+1){
            cout <<"NO"<<endl;
            return 0;
        }else{
            dp[av[i].second] = i + 2 - av[i].first;
        }
    }
    cout <<"YES"<<endl;
    for(long long int i = 1;i <= n-1;i++){
        cout << dp[i]<<endl;
    }
}
0