Submission #509007

#	Time	Username	Problem	Language	Result	Execution time	Memory
509007		Joo	Torrent (COI16_torrent)	C++17	69 / 100	737 ms	31392 KiB

torrent

#include <bits/stdc++.h>
using namespace std;

const int N = 3e5+10;

int st[2], dp[N], tmp[N], n;
vector<int> G[N], lis;

bool pre(int u, int p){
    if(u == st[1]){
        lis.emplace_back(u);
        return true;
    }
    for(int v : G[u]){
        if(v == p) continue;
        if(pre(v, u)){
            lis.emplace_back(u);
            return true;
        } 
    }
    return false;
}

void dfs(int u, int p, int ban){
    for(int v : G[u]) if(v != p and v != ban) dfs(v, u, ban);
    
    vector<int> vec;
    for(int v : G[u]) if(v != p and v != ban) vec.emplace_back(dp[v]);

    sort(vec.rbegin(), vec.rend());
    for(int i = 0; i < vec.size(); i++){
        dp[u] = max(dp[u], vec[i]+i+1);
    }
}

int main(void){
    ios_base::sync_with_stdio(0); cin.tie(0);
    cin >> n >> st[0] >> st[1];
    for(int i = 1; i < n; i++){
        int u, v; cin >> u >> v;
        G[u].emplace_back(v);
        G[v].emplace_back(u);
    }

    pre(st[0], -1);

    dfs(st[0], -1, lis.end()[-2]);
    dfs(st[1], -1, lis.end()[-1]);
    int ans = max(dp[st[0]], dp[st[1]]);

    int l = 0, r = (int)lis.size()-3;
    while(l <= r){
        int mid = (l+r)>>1;

        for(int i = 0; i <= n; i++) dp[i] = 0;
        assert(mid+1 <= (int)lis.size()-1);
        dfs(st[0], -1, lis[mid]);
        dfs(st[1], -1, lis[mid+1]);
        int res1 = max(dp[st[0]], dp[st[1]]);

        for(int i = 0; i <= n; i++) dp[i] = 0;
        assert(mid+2 <= (int)lis.size()-1);
        dfs(st[0], -1, lis[mid+1]);
        dfs(st[1], -1, lis[mid+2]);
        int res2 = max(dp[st[0]], dp[st[1]]);

        ans = min({ans, res1, res2});

        if(res1 < res2){
            r = mid-1;
        } else {
            l = mid+1;
        }
    }

    cout << ans << "\n";

    return 0;
}

Compilation message (stderr)

torrent.cpp: In function 'void dfs(int, int, int)':
torrent.cpp:31:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |     for(int i = 0; i < vec.size(); i++){
      |                    ~~^~~~~~~~~~~~

• Subtask 1: 0 / 31
• Subtask 2: 69 / 69