結果

問題 No.110 しましまピラミッド
ユーザー kichirb3kichirb3
提出日時 2018-03-21 19:02:09
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,902 bytes
コンパイル時間 979 ms
コンパイル使用メモリ 80,632 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-30 04:02:54
合計ジャッジ時間 2,209 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 1 ms
4,376 KB
testcase_10 AC 2 ms
4,380 KB
testcase_11 AC 1 ms
4,376 KB
testcase_12 AC 2 ms
4,376 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 1 ms
4,376 KB
testcase_15 AC 1 ms
4,376 KB
testcase_16 AC 1 ms
4,376 KB
testcase_17 AC 1 ms
4,376 KB
testcase_18 AC 2 ms
4,380 KB
testcase_19 AC 2 ms
4,376 KB
testcase_20 AC 1 ms
4,380 KB
testcase_21 AC 2 ms
4,376 KB
testcase_22 AC 1 ms
4,380 KB
testcase_23 AC 1 ms
4,376 KB
testcase_24 AC 1 ms
4,376 KB
testcase_25 AC 2 ms
4,376 KB
testcase_26 AC 1 ms
4,376 KB
testcase_27 AC 2 ms
4,376 KB
testcase_28 AC 2 ms
4,380 KB
testcase_29 AC 2 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// No.110 しましまピラミッド
// https://yukicoder.me/problems/no/110
//

#include <iostream>
#include <vector>
#include <algorithm>
#include <utility>
using namespace std;

int solve(vector<int> w_blocks, vector<int> b_blocks);


int main() {
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);

    unsigned int nw;
    cin >> nw;
    vector<int> w_blocks(nw);
    for (auto i = 0; i < nw; ++i)
        cin >> w_blocks[i];

    unsigned int nb;
    cin >> nb;
    vector<int> b_blocks(nb);
    for (auto i = 0; i < nb; ++i)
        cin >> b_blocks[i];

    int ans = solve(w_blocks, b_blocks);
    cout << ans << endl;
}

int solve(vector<int> w_blocks, vector<int> b_blocks) {
    w_blocks.push_back(-1);
    b_blocks.push_back(-1);
    sort(w_blocks.begin(), w_blocks.end());
    sort(b_blocks.begin(), b_blocks.end());

    const int WHITE = 0;
    const int BLACK = 1;
    vector<vector<int>> blocks(2);
    blocks[WHITE] = move(w_blocks);
    blocks[BLACK] = move(b_blocks);

    // 白いブロックを土台にする場合
    vector<int> ans_w;
    ans_w.push_back(blocks[WHITE][blocks[WHITE].size()-1]);
    int next_block = BLACK;
    while (true) {
        auto p = lower_bound(blocks[next_block].begin(), blocks[next_block].end(), ans_w[ans_w.size()-1]);
        if (*(p-1) == -1)
            break;
        ans_w.push_back(*(p-1));
        next_block = next_block == BLACK? WHITE: BLACK;
    }

    // 黒いブロックを土台にする場合
    vector<int> ans_b;
    ans_b.push_back(blocks[BLACK][blocks[BLACK].size()-1]);
    next_block = WHITE;
    while (true) {
        auto p = lower_bound(blocks[next_block].begin(), blocks[next_block].end(), ans_b[ans_b.size()-1]);
        if (*(p-1) == -1)
            break;
        ans_b.push_back(*(p-1));
        next_block = next_block == BLACK? WHITE: BLACK;
    }

    return max(ans_w.size(), ans_b.size());
}
0