C++の練習を兼ねて, AtCoder Regular Contest 044 の 問題D (suffix array) を解いてみた.
■感想.
1. 問題Dは, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 解説の方針で, 辞書順最小の文字列を抽出できることが, 非常に興味深く感じた.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Regular Contest 044 解説 を ご覧下さい.
■C++版プログラム(問題D／AC版).
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
// 解き直し. // https://www.slideshare.net/chokudai/arc044 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; #define repex(i, a, b, c) for(int i = a; i < b; i += c) #define repx(i, a, b) repex(i, a, b, 1) #define rep(i, n) repx(i, 0, n) #define repr(i, a, b) for(int i = a; i >= b; i--) int ia[1010101], oa[1010101], ans[1010101]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N){ scanf("%d", &ia[i + 1]); oa[ia[i + 1]] = i + 1; } // 2. 貪欲アルゴリズム. // ex. // 15 // 13 12 11 2 5 7 14 15 4 10 1 6 9 8 3 // // -> S16 の 存在を仮定して, 貪欲アルゴリズムをシュミレートしてみる. // ia(順列) 13 12 11 2 5 7 14 15 4 10 1 6 9 8 3 0 // oa(出現位置) 11 4 15 9 5 12 6 14 13 10 3 2 1 7 8 0 // // i S[a[i]] S[a[i + 1]] | S[a[i] + 1] S[a[i + 1] + 1] | cur // 1 S13 S12 | S14 (位置 7) > S13 (位置 1) | 0 -> 1 // 2 S12 S11 | S13 (位置 1) < S12 (位置 2) | 1 -> 1 // 3 S11 S2 | S12 (位置 2) < S3 (位置 15) | 1 -> 1 // 4 S2 S5 | S3 (位置 15) > S6 (位置 12) | 1 -> 2 // 5 S5 S7 | S6 (位置 12) < S8 (位置 14) | 2 -> 2 // 6 S7 S14 | S8 (位置 14) > S15 (位置 8) | 2 -> 3 // 7 S14 S15 | S15 (位置 8) > S16 (位置 0) | 3 -> 4 // 8 S15 S4 | S16 (位置 0) < S5 (位置 5) | 4 -> 4 // 9 S4 S10 | S5 (位置 5) > S11 (位置 3) | 4 -> 5 // 10 S10 S1 | S11 (位置 3) < S2 (位置 4) | 5 -> 5 // 11 S1 S6 | S2 (位置 4) < S7 (位置 6) | 5 -> 5 // 12 S6 S9 | S7 (位置 6) < S10 (位置 10) | 5 -> 5 // 13 S9 S8 | S10 (位置 10) < S9 (位置 13) | 5 -> 5 // 14 S8 S3 | S9 (位置 13) > S4 (位置 9) | 5 -> 6 // -> 以上から, FBGECFCFFFBBADE が 求める文字列と判明する. int cur = 0; repx(i, 1, N){ // S[a[i] + 1] > S[a[i + 1] + 1]. if(oa[ia[i] + 1] > oa[ia[i + 1] + 1]) ++cur; // 例外. if(cur >= 26){ puts("-1"); return 0; } // 設定. ans[ia[i + 1]] = cur; } // 3. 出力. repx(i, 1, N + 1) printf("%c%s", 'A' + ans[i], (i < N) ? "" : "\n"); return 0; } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
[入力例] 7 7 5 1 3 6 2 4 [出力例] ABACABA ※AtCoderのテストケースより [入力例] 12 7 1 9 10 2 4 3 6 12 5 11 8 [出力例] BDEDGFAHCCGG ※AtCoderのテストケースより [入力例] 3 1 2 3 [出力例] AAB [入力例] 15 11 3 10 6 4 7 12 1 15 8 2 5 9 13 14 [出力例] EGBCGCDFHCADHIF [入力例] 15 13 12 11 2 5 7 14 15 4 10 1 6 9 8 3 [出力例] FBGECFCFFFBBADE [入力例] 30 7 30 13 3 16 10 14 17 15 21 2 27 24 1 29 9 25 26 23 4 22 11 18 5 28 19 12 8 6 20 [出力例] HFBJLOANICKNBDECDLMOEKJGIJFMIB [入力例] 50 33 18 1 31 32 11 12 13 20 44 22 47 45 25 16 5 37 41 29 15 14 34 49 8 38 48 9 35 3 27 19 43 21 46 26 7 36 2 50 28 40 6 17 30 39 10 24 42 23 4 [出力例] APLUFROJKTCCCIIESANCNDTTENMQHSBCAILPGJSRGTNDENEKIQ [入力例] 75 6 7 8 9 10 1 2 3 26 27 28 4 5 11 12 64 65 66 13 32 33 34 35 71 72 73 74 75 14 15 31 36 37 38 39 40 49 50 51 52 56 57 58 59 60 61 62 46 47 48 63 53 54 55 67 68 69 21 22 23 24 25 16 17 18 19 20 41 42 43 44 45 70 29 30 [出力例] BBBDEAAAAAEEGJJSSSSTRRRRSCCCVWKHHHHKKKKKTTTTUNNOLLLLPPQMMMMMMMPFFFQQQVIIIIJ [入力例] 100 89 70 88 35 9 92 96 54 10 38 94 87 56 53 39 46 67 23 43 66 41 26 63 55 75 51 6 25 21 68 73 27 95 62 42 80 15 33 85 32 14 30 3 49 64 76 7 91 16 45 72 77 100 24 69 58 22 11 81 71 47 99 1 2 52 18 5 74 8 50 36 60 13 17 65 93 82 37 44 40 57 4 59 98 84 90 12 86 31 78 83 34 28 79 61 19 29 97 48 20 [出力例] -1 [入力例] 200 1 2 3 4 5 6 7 8 9 20 11 12 13 14 15 16 17 18 19 10 21 22 23 24 25 26 27 28 29 40 31 32 33 34 35 36 37 38 39 30 41 42 43 44 45 46 47 48 49 60 51 52 53 54 55 56 57 58 59 50 61 62 63 64 65 66 67 68 69 80 71 72 73 74 75 76 77 78 79 70 81 82 83 84 85 86 87 88 89 100 91 92 93 94 95 96 97 98 99 90 101 102 103 104 105 106 107 108 109 120 111 112 113 114 115 116 117 118 119 110 121 122 123 124 125 126 127 128 129 140 131 132 133 134 135 136 137 138 139 130 141 142 143 144 145 146 147 148 149 160 151 152 153 154 155 156 157 158 159 150 161 162 163 164 165 166 167 168 169 180 171 172 173 174 175 176 177 178 179 170 181 182 183 184 185 186 187 188 189 200 191 192 193 194 195 196 197 198 199 190 [出力例] AAAAAAAAACBBBBBBBBCACCCCCCCCCEDDDDDDDDECEEEEEEEEEGFFFFFFFFGEGGGGGGGGGIHHHHHHHHIGIIIIIIIIIKJJJJJJJJKIKKKKKKKKKMLLLLLLLLMKMMMMMMMMMONNNNNNNNOMOOOOOOOOOQPPPPPPPPQOQQQQQQQQQSRRRRRRRRSQSSSSSSSSSUTTTTTTTTUT |
■参照サイト
AtCoder Regular Contest 044