/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 82ms 99.012 MiB
#2 Accepted 82ms 99.02 MiB
#3 Accepted 83ms 98.996 MiB
#4 Accepted 83ms 98.934 MiB
#5 Accepted 403ms 124.398 MiB
#6 Accepted 439ms 124.414 MiB
#7 Accepted 465ms 124.566 MiB
#8 Accepted 973ms 124.664 MiB
#9 Accepted 352ms 124.559 MiB
#10 Accepted 1382ms 124.465 MiB
#11 Accepted 1737ms 124.461 MiB
#12 Accepted 1916ms 124.469 MiB
#13 Accepted 1265ms 123.289 MiB
#14 Accepted 286ms 123.281 MiB
#15 Accepted 448ms 123.289 MiB
#16 Accepted 1146ms 123.297 MiB
#17 Accepted 318ms 123.293 MiB
#18 Accepted 1080ms 123.289 MiB
#19 Accepted 948ms 122.863 MiB
#20 Accepted 830ms 138.141 MiB
#21 Accepted 571ms 153.438 MiB
#22 Wrong Answer 18ms 18.66 MiB
#23 Wrong Answer 18ms 19.152 MiB

Code

// #pragma GCC optimize("Ofast")
// #pragma GCC target("avx2")
#include<bits/stdc++.h>
// #include <ext/pb_ds/assoc_container.hpp>
// #include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
// #define int long long
#define endl '\n'

using namespace std;
using pii = pair<int, int>;
using tup = tuple<int, int, int>;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
// template <class T> using ordered_set = tree<T, null_type,
//                          less_equal<T>, rb_tree_tag, tree_order_statistics_node_update>;

const int inf = 1e9;
const int mod = 1000000007;
const double eps = 1e-9;
const int N = 200005;

void preprocess() {}

vector<int> e[N];
multiset<int> mst[N];
int sz[N], mx[N], par[N];

void dfs(int u, int p) {
    sz[u] = 1;
    mx[u] = 1;
    par[u] = p;
    for(int v : e[u]) if(v != p) {
        dfs(v, u);
        sz[u] += sz[v];
        mx[u] = max(mx[v] + 1, mx[u]);
    }
}

int dp[N][105];
vector<int> cst[N];

int DP(int i, int k, int p) {
    if(dp[i][k] != -1) return dp[i][k];
    // if(sz[i] <= k+1) return dp[i][k] = sz[i];

    int ans = 1;
    for(int jj=0; jj<e[i].size(); jj++) {
        int j = e[i][jj];
        if(j == p) continue;
        ans = max(ans, 1 + DP(j, k, i));
        if(k) {
            int mxnow = cst[i][jj];
            int lo = 0, hi = k-1;

            int now = 0;
            while(lo <= hi) {
                int mid = lo + hi >> 1;

                int f1 = 1 + DP(j, mid, i) + min(mxnow, k - mid);
                int f2 = 1 + DP(j, mid+1, i) + min(mxnow, k - (mid+1));

                if(f1 <= f2) lo = mid + 1;
                else hi = mid - 1;

                now = max({now, f1, f2});
            }
            ans = max(ans, now);
        }
    }

    return dp[i][k] = ans;
}

void solve(int tc) {
    int n, k;
    cin >> n >> k;

    if(n <= k+1) {
        cout << n << endl;
        return;
    }

    for(int i=1; i<n; i++) {
        int u, v;
        cin >> u >> v;
        e[u].push_back(v);
        e[v].push_back(u);
    }

    memset(dp, -1, sizeof dp);
    dfs(1, -1);

    for(int i=1; i<=n; i++) {
        for(int j : e[i]) if(j - par[i]) {
            mst[i].insert(mx[j]);
        }

        for(int j : e[i]) {
            if(j == par[i]) {
                cst[i].push_back(0);
                continue;
            }
            mst[i].erase(mst[i].find(mx[j]));
            if(mst[i].size()) {
                cst[i].push_back(*mst[i].rbegin());
            }
            else cst[i].push_back(0);
            mst[i].insert(mx[j]);
        }
    }
    cout << DP(1, k, -1) << endl;
}   

int32_t main() {
    cin.tie(NULL)->sync_with_stdio(false);
    cout.precision(10);

    preprocess();

    int T = 1;
    // cin >> T;

    for (int i = 1; i <= T; i++) {  
        solve(i);
    }

    return 0;
}

Information

Submit By
Type
Submission
Problem
P1111 Thakurs tree game
Contest
Brain Booster #7
Language
C++17 (G++ 13.2.0)
Submit At
2024-11-05 16:42:43
Judged At
2024-11-11 02:26:34
Judged By
Score
97
Total Time
1916ms
Peak Memory
153.438 MiB