結果

問題 No.1215 都市消滅ビーム
ユーザー chocorusk
提出日時 2020-09-07 11:53:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,565 bytes
コンパイル時間 1,756 ms
コンパイル使用メモリ 146,520 KB
最終ジャッジ日時 2025-01-14 08:10:14
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 35 WA * 5
権限があれば一括ダウンロードができます

ソースコード

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>=1; 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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0