結果

問題 No.1488 Max Score of the Tree
ユーザー jutama
提出日時 2021-04-23 23:49:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 148 ms / 2,000 ms
コード長 4,354 bytes
コンパイル時間 6,887 ms
コンパイル使用メモリ 185,736 KB
最終ジャッジ日時 2025-01-21 00:30:08
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <bitset>
#include <fstream>
#include <functional>
#include <iostream>
#include <iomanip>
#include <limits>
#include <map>
#include <math.h>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <unordered_map>
#include <unordered_set>
#include <vector>
using namespace std;
using namespace atcoder;

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<ll, ld> pid;
typedef pair<ll, string> pls;
typedef vector<bool> vb;
typedef vector<vector<bool>> vvb;
typedef vector<int> vi;
typedef vector<vector<int>> vvi;
typedef vector<vector<vector<int>>> vvvi;
typedef vector<vector<vector<vector<int>>>> vvvvi;
typedef vector<ll> vl;
typedef vector<vector<ll>> vvl;
typedef vector<vector<vector<ll>>> vvvl;
typedef vector<vector<vector<vector<ll>>>> vvvvl;
typedef vector<ld> vd;
typedef vector<vector<ld>> vvd;
typedef vector<string> vs;
#define rep(i,n) for(auto i=0; i<n; ++i)
#define repm(i,s,n) for(auto i=s; i<n; ++i)
#define repd(i,n) for(auto i=n-1; i>=0; --i)
#define repdm(i,e,n) for(auto i=n-1; i>=e; --i)
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()

template <class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template <class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }

template <class mint, internal::is_modint_t<mint>* = nullptr>
constexpr istream& operator>>(istream& is, mint& x) noexcept {long long v = 0; std::cin >> v; x = v; return is;}
template <class mint, internal::is_modint_t<mint>* = nullptr>
constexpr ostream& operator<<(ostream& os, const mint& x) noexcept {os << x.val(); return os;}
inline void _n() { std::cout << std::endl; }
template <class T> inline void _(const T a) { std::cout << a; }
template <class T> inline void _l(const T a) { _(a); _n(); }
template <class T> inline void _s(const T a) { _(a); _(' '); }
template <class T> inline void _v(const std::vector<T> v) { for(auto a : v) _(a); }
template <class T> inline void _vl(const std::vector<T> v) { for(auto a : v) _l(a); }
template <class T> inline void _vs(const std::vector<T> v) { for(auto a : v) _s(a); _n(); }
template <class T> inline void _vvl(const std::vector<std::vector<T>> v) { for(auto a : v) { _v(a); _n(); } }
template <class T> inline void _vvs(const std::vector<std::vector<T>> v) { for(auto a : v) { _vs(a); } }
inline void ynl(const bool b) {_l(b ? "Yes" : "No");}
inline void ynu(const bool b) {_l(b ? "YES" : "NO");}

const int INF = 1 << 30;
const long long INF_LL = 1LL << 62;
const long long MOD1 = 1000000007;
const long long MOD9 = 998244353;
using mint1 = modint1000000007;
using mint9 = modint998244353;

//* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ *//
struct edge {
    int idx;
    int to;
    ll cost;
};

int N;
int K;
vector<vector<edge>> edges;
vector<ll> e_cnt;
vector<ll> e_cost;
vector<int> visit;

void input() {
   
    cin >> N >> K;
    edges.resize(N, vector<edge>());
    e_cnt.resize(N-1);
    e_cost.resize(N-1);
    visit.resize(N);
    rep(i, N-1) {
        int a, b; cin >> a >> b; a--; b--;
        ll c; cin >> c;
        edges[a].push_back({i, b, c});
        edges[b].push_back({i, a, c});
        e_cost[i] = c;
    }
    
}

int dfs(int v, int ei = -1) {
    
    int cnt = 0;
    bool leaf = 1;
    visit[v] = 1;
    
    for(auto e : edges[v]) {
        if(!visit[e.to]) {
            leaf = 0;
            cnt += dfs(e.to, e.idx);
        }
    }
    
    if(ei != -1) {
        e_cnt[ei] = cnt + leaf;
    }
    
    return cnt + leaf;
}


void solve() {
    
    dfs(0);
    // _vs(e_cnt);
    
    ll ans = 0;
    vvl dp(N, vl(K+1));
    rep(i, N-1) {
        ans += e_cost[i] * e_cnt[i];
        rep(j, K+1) {
            chmax(dp[i+1][j], dp[i][j]);
            if(j + e_cost[i] <= K) chmax(dp[i+1][j + e_cost[i]], dp[i][j] + e_cost[i] * e_cnt[i]);
        }
    }
    
    ll tmp = 0;
    rep(j, K+1) {
        chmax(tmp, dp[N-1][j]);
    }
    ans += tmp;
    _l(ans);
    
    
    
    
}

//* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ *//

int main() {
    
    std::ifstream in("input.txt");
    std::cin.rdbuf(in.rdbuf());
    std::cin.tie(0);
    ios::sync_with_stdio(false);
    
    input();
    solve();
    
    return 0;
}
0