結果
問題 | No.1373 Directed Operations |
ユーザー |
![]() |
提出日時 | 2021-11-01 20:24:14 |
言語 | C (gcc 13.3.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,568 bytes |
コンパイル時間 | 553 ms |
コンパイル使用メモリ | 32,116 KB |
実行使用メモリ | 13,640 KB |
最終ジャッジ日時 | 2024-10-10 11:41:37 |
合計ジャッジ時間 | 5,251 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 1 |
other | TLE * 1 -- * 18 |
ソースコード
#include<stdio.h>int h[100005], l;int A[100005];int comp_h(int a, int b){if (A[h[a]] < A[h[b]])return 1;elsereturn -1;}void swap_h(int a, int b){int f = h[a];h[a] = h[b];h[b] = f;return;}void push(int ne){h[l] = ne;int p = l;l++;for (; p > 0; p = (p - 1) / 2)if (comp_h((p - 1) / 2, p) > 0)swap_h((p - 1) / 2, p);return;}int pop(){l--;swap_h(0, l);int p = 0;for (;;){if (2 * p + 2 < l){if (comp_h(2 * p + 1, 2 * p + 2) > 0){if (comp_h(p, 2 * p + 2) > 0)swap_h(p, 2 * p + 2);p = 2 * p + 2;}else{if (comp_h(p, 2 * p + 1) > 0)swap_h(p, 2 * p + 1);p = 2 * p + 1;}}else if (2 * p + 1 < l){if (comp_h(p, 2 * p + 1) > 0)swap_h(p, 2 * p + 1);p = 2 * p + 1;}elsebreak;}return h[l];}int p[100005];int root(int n){if (p[n] != n)p[n] = root(p[n]);return p[n];}void uni(int x, int y){x = root(x);y = root(y);p[x] = y;}int main(){int n;scanf("%d", &n);int i, j;for (i = 0; i < n - 1; i++)scanf("%d", &A[i]);l = 0;for (i = 0; i < n - 1; i++)push(i);int c[100005];for (i = 0; i < n - 1; i++)c[i] = pop();int ans[100005];for (i = 0; i < n; i++)p[i] = i;for (i = 0; i < n - 1; i++){j = n - A[c[i]] - 1;while (root(j) == root(j + A[c[i]])){j--;if (j < 0)break;}if (j < 0){printf("NO\n");return 0;}ans[c[i]] = j + 1;uni(j, j + A[c[i]]);}printf("YES\n");for (i = 0; i < n - 1; i++)printf("%d\n", ans[i]);return 0;}