#836251 (C++17) No.417 チューリップバブル

提出ソース

結果

問題	No.417 チューリップバブル
ユーザー	zawakasu
提出日時	2023-01-31 11:45:42
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	318 ms / 2,000 ms
コード長	3,827 bytes
コンパイル時間	2,319 ms
コンパイル使用メモリ	215,408 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-06-30 15:56:16
合計ジャッジ時間	7,060 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	5 ms 6,940 KB
testcase_09	AC	9 ms 6,940 KB
testcase_10	AC	12 ms 6,944 KB
testcase_11	AC	46 ms 6,940 KB
testcase_12	AC	47 ms 6,940 KB
testcase_13	AC	18 ms 6,944 KB
testcase_14	AC	76 ms 6,940 KB
testcase_15	AC	6 ms 6,944 KB
testcase_16	AC	7 ms 6,944 KB
testcase_17	AC	44 ms 6,944 KB
testcase_18	AC	42 ms 6,944 KB
testcase_19	AC	41 ms 6,940 KB
testcase_20	AC	161 ms 6,944 KB
testcase_21	AC	161 ms 6,944 KB
testcase_22	AC	154 ms 6,944 KB
testcase_23	AC	161 ms 6,944 KB
testcase_24	AC	2 ms 6,940 KB
testcase_25	AC	159 ms 6,940 KB
testcase_26	AC	18 ms 6,940 KB
testcase_27	AC	110 ms 6,940 KB
testcase_28	AC	157 ms 6,940 KB
testcase_29	AC	159 ms 6,944 KB
testcase_30	AC	156 ms 6,944 KB
testcase_31	AC	158 ms 6,940 KB
testcase_32	AC	1 ms 6,944 KB
testcase_33	AC	5 ms 6,940 KB
testcase_34	AC	32 ms 6,940 KB
testcase_35	AC	318 ms 6,940 KB
testcase_36	AC	311 ms 6,944 KB
testcase_37	AC	308 ms 6,944 KB
testcase_38	AC	308 ms 6,944 KB
testcase_39	AC	308 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>

#define all(x) begin(x), end(x)
#define times(x) for (int _ = 0 ; _ < (int)(x) ; _++)

using i32 = int;
using i64 = long long;
using ld = long double;
using usize = std::size_t;

template <class T1, class T2>
inline bool chmax(T1 &a, const T2 &b) { return a < b and (a = b, true); }
template <class T1, class T2>
inline bool chmin(T1 &a, const T2 &b) { return a > b and (a = b, true); }

constexpr i64 supl = (std::numeric_limits<i64>::max() >> 1) - 100;
constexpr i32 supi = (std::numeric_limits<i32>::max() >> 1) - 100;



namespace zawa::input {

template <typename T> 
void in(T& res) { std::cin >> res; }

template <typename Head, typename... Tail>
void in(Head& head, Tail&... tail) { in(head); in(tail...); }

template <typename T, typename U>
void in(std::pair<T, U>& res) { in(res.first); in(res.second); }

template <typename T>
void in(std::vector<T>& res) { for (auto& r : res) { in(r); } }

} // namespace zawa::input
using zawa::input::in;



namespace zawa::output {

void out() { 
    std::cout << std::endl; 
}

template <class T>
void out(const T& a) { 
    std::cout << a << std::endl; 
}

template <class T>
void out(const std::vector<T>& as) { 
    for (std::size_t i = 0 ; i < as.size() ; i++) { 
        std::cout << as[i] << (i + 1 == as.size() ? '\n' : ' '); 
    } 
}

template <class Head, class... Tail>
void out(const Head& head, const Tail&... tail) { 
    std::cout << head; 
    if (sizeof...(tail)) { 
        std::cout << ' '; 
    } 
    out(tail...);
}

void yesno(bool flag, std::string yes = "Yes", std::string no = "No") {
    std::cout << (flag ? yes : no) << std::endl;
}

} // namespace zawa::output
using zawa::output::out;
using zawa::output::yesno;

// #include "atcoder/modint"
// using mint = atcoder::modint998244353;
// using mint = atcoder::modint1000000007;

// #include "src/template/accum1d.hpp"
// #include "src/template/binary-search.hpp"
// #include "src/template/binary-search-ld.hpp"
// #include "src/algorithm/compression.hpp"
// #include "src/algorithm/RLE.hpp"

// #include "src/graph/Read-Graph.hpp"


namespace zawa {

template <typename T>
std::vector<std::vector<std::pair<int, T>>> read_weighted_graph(int n, int m, bool undirect = true, bool minus = true) {
    std::vector<std::vector<std::pair<int, T>>> res(n, std::vector(0, std::pair<int, T>()));
    for (int _ = 0 ; _ < m ; _++) {
        int u, v; std::cin >> u >> v;
        T c; std::cin >> c;
        res[u - minus].emplace_back(v - minus, c);
        if (undirect) {
            res[v - minus].emplace_back(u - minus, c);
        }
    }
    return res;
}

template <typename T>
std::vector<std::vector<std::pair<int, T>>> read_weighted_tree(int n, bool undirect = true) {
    return read_weighted_graph<T>(n, n - 1, undirect);
}

} // namespace zawa

using namespace std;

void main_() {
	i32 n, m; in(n, m);
	vector us(n, 0); in(us);
	vector G(n, vector(0, pair(0, 0)));
	times(n - 1) {
		i32 u, v, c; in(u, v, c);
		G[u].emplace_back(v, c);
		G[v].emplace_back(u, c);
	}
	vector dp(n, vector(m + 1, -supl));
	auto rec = [&](auto rec, i32 v, i32 p) -> void {
		for (usize i = 0 ; i < G[v].size() ; i++) {
			if (G[v][i].first == p) {
				G[v].erase(G[v].begin() + i);
				break;
			}
		}
		for (auto [x, _] : G[v]) {
			rec(rec, x, v);
		}
		dp[v][0] = us[v];
		for (auto [x, c] : G[v]) {
			vector nxt = dp[v];
			for (i32 i = 0 ; i <= m ; i++) {
				if (dp[v][i] == -supi) {
					continue;
				}
				for (i32 j = 0 ; j + i + 2 * c <= m ; j++) {
					if (dp[x][j] == -supi) {
						continue;
					}
					chmax(nxt[j + i + 2 * c], dp[v][i] + dp[x][j]);	
				}
			}
			dp[v] = move(nxt);
		}
	};
	rec(rec, 0, -1);
	out(*max_element(all(dp[0])));
}

i32 main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout << fixed << setprecision(10);

	main_();

	return 0;
}

yukicoder

結果

テストケース

ソースコード