結果

問題 No.1215 都市消滅ビーム
ユーザー chocoruskchocorusk
提出日時 2020-09-07 11:59:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,769 ms / 6,000 ms
コード長 4,565 bytes
コンパイル時間 1,849 ms
コンパイル使用メモリ 151,668 KB
実行使用メモリ 74,512 KB
最終ジャッジ日時 2024-05-06 23:32:13
合計ジャッジ時間 21,807 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 3 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 3 ms
6,940 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 300 ms
22,048 KB
testcase_14 AC 405 ms
23,368 KB
testcase_15 AC 1,053 ms
46,232 KB
testcase_16 AC 237 ms
18,704 KB
testcase_17 AC 419 ms
31,900 KB
testcase_18 AC 445 ms
28,988 KB
testcase_19 AC 37 ms
7,644 KB
testcase_20 AC 808 ms
41,336 KB
testcase_21 AC 15 ms
6,948 KB
testcase_22 AC 614 ms
33,304 KB
testcase_23 AC 565 ms
36,800 KB
testcase_24 AC 512 ms
27,080 KB
testcase_25 AC 311 ms
24,576 KB
testcase_26 AC 603 ms
38,236 KB
testcase_27 AC 680 ms
45,468 KB
testcase_28 AC 379 ms
22,860 KB
testcase_29 AC 373 ms
22,564 KB
testcase_30 AC 219 ms
28,200 KB
testcase_31 AC 134 ms
15,044 KB
testcase_32 AC 93 ms
22,568 KB
testcase_33 AC 81 ms
19,504 KB
testcase_34 AC 1,769 ms
74,240 KB
testcase_35 AC 1,743 ms
74,236 KB
testcase_36 AC 1,735 ms
74,232 KB
testcase_37 AC 1,687 ms
74,384 KB
testcase_38 AC 1,575 ms
74,512 KB
testcase_39 AC 1,719 ms
74,384 KB
testcase_40 AC 3 ms
6,940 KB
testcase_41 AC 2 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#define popcount __builtin_popcount
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
struct LCA{
    vector<vector<int>> g;
    vector<int> d;
    vector<vector<int>> p;
    int log;
    int n;
    LCA(const vector<vector<int>> &g):n(g.size()), g(g), d(g.size()){
        log=0;
        while(1<<log<=n) log++;
        p.resize(log, vector<int>(n));
    }
    void dfs(int x, int prev){
        for(auto y:g[x]){
            if(y==prev) continue;
            d[y]=d[x]+1;
            p[0][y]=x;
            dfs(y, x);
        }
    }
    void build(){
        dfs(0, -1);
        for(int i=1; i<log; i++){
            for(int j=0; j<n; j++){
                p[i][j]=p[i-1][p[i-1][j]];
            }
        }
    }
    int lca(int a, int b){
        if(d[a]>d[b]) swap(a, b);
        int dd=d[b]-d[a], i=0;
        int a1=a, b1=b;
        while(dd){
            if(dd&1) b1=p[i][b1];
            dd>>=1;
            i++;
        }
        if(a1==b1) return a1;
        for(int j=log-1; j>=0; j--){
            if(p[j][a1]!=p[j][b1]){
                a1=p[j][a1], b1=p[j][b1];
            }
        }
        return p[0][a1];
    }
    int dist(int a, int b){
        return d[a]+d[b]-2*d[lca(a, b)];
    }
};
int main()
{
    int n, k; cin>>n>>k;
    int c[100010];
    ll w[100010];
    for(int i=0; i<k; i++){
        cin>>c[i];
        c[i]--;
    }
    for(int i=0; i<k; i++){
        cin>>w[i];
    }
    vector<vector<int>> g(n);
    for(int i=0; i<n-1; i++){
        int a, b; cin>>a>>b;
        a--; b--;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    LCA lca(g);
    lca.build();
    int p0=lca.lca(c[0], c[k-1]);
    int d0=lca.d[p0];
    int cl[100010], cr[100010];
    cl[0]=c[0], cr[k-1]=c[k-1];
    for(int i=1; i<k; i++) cl[i]=lca.lca(c[i], cl[i-1]);
    int dr[100010]; dr[k-1]=lca.d[c[k-1]];
    for(int i=k-2; i>=0; i--){
        cr[i]=lca.lca(c[i], cr[i+1]);
        dr[i]=lca.d[cr[i]];
    }
    ll sl[100010], sr[100010], sr1[100010];
    sl[0]=w[0];
    for(int i=1; i<k; i++) sl[i]=sl[i-1]+w[i];
    sr[k-1]=w[k-1], sr1[k-1]=sr[k-1]+lca.d[cr[k-1]];
    for(int i=k-2; i>=0; i--){
        sr[i]=sr[i+1]+w[i];
        sr1[i]=sr[i]+lca.d[cr[i]];
    }
    int sz=1;
    while(sz<k) sz<<=1;
    vector<vector<ll>> seg(2*sz), seg1(2*sz);
    for(int i=0; i<k; i++){
        int t=i+sz;
        seg[t].push_back(sr[i]);
        seg1[t].push_back(sr1[i]);
        while(t>1){
            t>>=1;
            seg[t].push_back(sr[i]);
            seg1[t].push_back(sr1[i]);
        }
    }
    for(int i=1; i<2*sz; i++){
        sort(seg[i].begin(), seg[i].end());
        sort(seg1[i].begin(), seg1[i].end());
    }
    auto query=[&](int l, int r, ll x){
        l+=sz, r+=sz;
        int ret=0;
        for(;l<r; l>>=1, r>>=1){
            if(r&1){
                r--;
                ret+=upper_bound(seg[r].begin(), seg[r].end(), x)-seg[r].begin();
            }
            if(l&1){
                ret+=upper_bound(seg[l].begin(), seg[l].end(), x)-seg[l].begin();
                l++;
            }
        }
        return ret;
    };
    auto query1=[&](int l, int r, ll x){
        l+=sz, r+=sz;
        int ret=0;
        for(;l<r; l>>=1, r>>=1){
            if(r&1){
                r--;
                ret+=upper_bound(seg1[r].begin(), seg1[r].end(), x)-seg1[r].begin();
            }
            if(l&1){
                ret+=upper_bound(seg1[l].begin(), seg1[l].end(), x)-seg1[l].begin();
                l++;
            }
        }
        return ret;
    };

    ll l=-1.1e14, r=1.1e14;
    while(r-l>1){
        ll m=(l+r)/2;
        ll cnt=1;
        for(int i=0; i<k-1; i++){
            if(sl[i]+lca.d[cl[i]]<=m) cnt++;
            int d1=min(lca.d[cl[i]], d0);
            int t=lower_bound(dr, dr+k, d1)-dr;
            cnt+=query(max(t, i+2), k, m-d1-sl[i]);
            if(i+2<t){
                cnt+=query1(i+2, t, m-sl[i]);
            }
        }
        for(int i=k-1; i>=0; i--) if(sr[i]+lca.d[cr[i]]<=m) cnt++;
        ll tot=(ll)k*(k+1)/2+1;
        if(cnt>=(tot+1)/2) r=m;
        else l=m;
    }
    cout<<r<<endl;
    return 0;
}
0