結果
問題 | No.2161 Black Market |
ユーザー | suisen |
提出日時 | 2022-12-24 03:53:04 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,886 bytes |
コンパイル時間 | 1,325 ms |
コンパイル使用メモリ | 99,704 KB |
実行使用メモリ | 8,720 KB |
最終ジャッジ日時 | 2024-11-18 05:07:22 |
合計ジャッジ時間 | 3,013 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | WA | - |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 3 ms
5,248 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 2 ms
5,248 KB |
testcase_12 | AC | 2 ms
5,248 KB |
testcase_13 | AC | 2 ms
5,248 KB |
testcase_14 | AC | 2 ms
5,248 KB |
testcase_15 | WA | - |
testcase_16 | AC | 2 ms
5,248 KB |
testcase_17 | AC | 2 ms
5,248 KB |
testcase_18 | AC | 2 ms
5,248 KB |
testcase_19 | AC | 2 ms
5,248 KB |
testcase_20 | AC | 34 ms
8,592 KB |
testcase_21 | AC | 34 ms
8,600 KB |
testcase_22 | AC | 36 ms
8,592 KB |
testcase_23 | AC | 43 ms
8,588 KB |
testcase_24 | AC | 114 ms
8,596 KB |
testcase_25 | AC | 39 ms
7,656 KB |
testcase_26 | AC | 101 ms
8,720 KB |
testcase_27 | AC | 3 ms
5,248 KB |
testcase_28 | AC | 5 ms
5,248 KB |
testcase_29 | AC | 5 ms
5,248 KB |
testcase_30 | WA | - |
testcase_31 | WA | - |
testcase_32 | AC | 2 ms
5,248 KB |
testcase_33 | WA | - |
testcase_34 | AC | 3 ms
5,248 KB |
testcase_35 | AC | 2 ms
5,248 KB |
testcase_36 | AC | 8 ms
5,248 KB |
testcase_37 | AC | 5 ms
5,248 KB |
testcase_38 | AC | 23 ms
5,248 KB |
testcase_39 | AC | 4 ms
5,248 KB |
ソースコード
#include <algorithm> #include <iostream> #include <numeric> #include <vector> #include <atcoder/fenwicktree> std::vector<std::vector<std::pair<int64_t, int64_t>>> enumerate_all( const size_t N, const size_t K, const std::vector<int64_t> w, const std::vector<int64_t> v ) { std::vector<std::vector<std::pair<int64_t, int64_t>>> res(K + 1); res[0].emplace_back(0, 0); for (size_t i = 0; i < N; ++i) { std::vector<std::size_t> mid(K + 1); for (size_t j = 1; j <= K; ++j) { mid[j] = res[j].size(); } for (size_t j = K; j --> 0;) { for (const auto& [wsum, vsum] : res[j]) { res[j + 1].emplace_back(wsum + w[i], vsum + v[i]); } } for (size_t j = 1; j <= K; ++j) { std::inplace_merge(res[j].begin(), res[j].begin() + mid[j], res[j].end()); } } return res; } int64_t solve( const size_t N, const size_t K, const int64_t W, const int64_t V, const std::vector<int64_t> w, const std::vector<int64_t> v ) { const size_t H = N / 2; auto L = enumerate_all( H, K, std::vector<int64_t>(w.begin(), w.begin() + H), std::vector<int64_t>(v.begin(), v.begin() + H) ); auto R = enumerate_all( N - H, K, std::vector<int64_t>(w.begin() + H, w.end()), std::vector<int64_t>(v.begin() + H, v.end()) ); int64_t ans = 0; for (size_t j = 0; j <= std::min(K, H); ++j) { std::vector<int64_t> comp_v; for (const auto &e : R[j]) { comp_v.emplace_back(e.second); } std::sort(comp_v.begin(), comp_v.end()); comp_v.erase(std::unique(comp_v.begin(), comp_v.end()), comp_v.end()); for (auto &e : R[j]) { e.second = std::lower_bound(comp_v.begin(), comp_v.end(), e.second) - comp_v.begin(); } const size_t m = comp_v.size(); for (size_t i = 0; i <= std::min(K - j, N - H); ++i) { atcoder::fenwick_tree<uint32_t> ft(m); auto itR = R[j].begin(); for (auto itL = L[i].rbegin(); itL != L[i].rend(); ++itL) { auto [wsum, vsum] = *itL; int64_t maxw = W - wsum; for (; itR != R[j].end() and itR->first <= maxw; ++itR) { ft.add(itR->second, 1); } int64_t maxv = V - vsum; std::size_t l = std::lower_bound(comp_v.begin(), comp_v.end(), maxv) - comp_v.begin(); ans += (itR - R[j].begin()) - ft.sum(0, l); } } } return ans; } int main() { size_t N, K; int64_t W, V; std::cin >> N >> K >> W >> V; std::vector<int64_t> w(N), v(N); for (size_t i = 0; i < N; ++i) { std::cin >> w[i] >> v[i]; } std::cout << solve(N, K, W, V, w, v) << std::endl; }