C++の練習を兼ねて, AtCoder Regular Contest 031 の 問題C (C – 積み木) を解いてみた.
■感想.
1. 解答方針が見えなかったので, 解説を参照して実装したところ, ようやくAC版となった.
2. Binary Indexed Tree の 復習が出来たので, 非常に良かったと思う.
※ 公式のライブラリを拝借させて頂いてます.
3. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイトARC 031 解説をご覧下さい.
■C++版プログラム(問題C／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 67 68 69 70 71 72 |
// 解き直し. // https://www.slideshare.net/chokudai/arc031 // C++14 (GCC 5.4.1) #include <bits/stdc++.h> using namespace std; using LL = long long; #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--) #define pb push_back const int MAX = 101010; int ih[MAX], oh[MAX]; // Binary Indexed Tree (Fenwick Tree) // https://youtu.be/lyHk98daDJo?t=7960 template<typename T> struct BIT{ int n; vector<T> d; BIT(int n = 0) : n(n), d(n + 1) {} void add(int i, T x = 1){ for(i++; i <= n; i += i & -i) d[i] += x; } T sum(int i){ T x = 0; for(i++; i; i -= i & -i) x += d[i]; return x; } T sum(int l, int r){ return sum(r - 1) - sum(l - 1); } }; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N){ scanf("%d", &ih[i]); oh[ih[i]] = i; } // 2. 解説通り. BIT<LL> d(N); rep(i, N) d.add(i, 1); // 3. 積み木を調べる. // ex. // d.sum(0, 5) -> 区間[l + 1, r] = [1, 5] の 和と見て, 合計 5 と理解. LL ans = 0; rep(i, N - 1){ // 左右の和を抽出. LL l = d.sum(0, oh[i + 1]); LL r = d.sum(oh[i + 1] + 1, N); // 小さい方を加算. ans += min(l, r); // printf("i=%d l=%lld r=%lld\n", i, l, r); // Binary Indexed Tree を 更新. d.add(oh[i + 1], -1); // rep(j, N) printf("%lld ", d.sum(j, j + 1)); // puts(""); } // 4. 出力. printf("%lld\n", ans); 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 |
[入力例] 4 2 4 1 3 [出力例] 1 ※AtCoderのテストケースより [入力例] 5 2 4 1 3 5 [出力例] 3 ※AtCoderのテストケースより [入力例] 6 1 2 4 3 5 6 [出力例] 1 ※AtCoderのテストケースより [入力例] 6 6 5 2 1 3 4 [出力例] 5 [入力例] 12 11 3 4 7 8 10 12 1 2 5 6 9 [出力例] 15 [入力例] 25 22 15 13 11 25 3 20 4 14 7 21 8 24 18 10 12 17 1 2 5 6 23 9 16 19 [出力例] 78 [入力例] 32 22 30 15 13 11 29 25 3 20 4 26 14 7 21 8 24 18 28 32 10 12 17 1 2 31 5 6 23 9 16 19 27 [出力例] 130 |
■参照サイト
AtCoder Regular Contest 031
Binary Indexed Tree (Fenwick Tree)