cogs2109 [NOIP2015] 运输计划

cogs2109 [NOIP2015] 运输计划


二分答案+树上差分。

STO链剖巨佬们我不会(太虚伪了吧

首先二分一个答案,下界为0,上界为max{路径长度}。

然后判断一个答案是否可行,这里用到树上差分。

(阔以理解为前缀和???)

随便搞出所有路径的LCA。

倍增可能会MLE,Trajan(没拼错)不会,只会链剖求LCA。

对于一个mid,所有路径>mid的路径都需要减一个值。设有k条。这个排序后二分求(mdzz二分写跪N多次)

由于只能

把一条路权值置0,这条路肯定要在k条路的交上,而且一定是权值最大的那个

如何搞出k条路的交?

树剖+树状数组,每次把路上所有边权++,然后检查那些边权是k的边,选一个最大怎么可能= =(虽然可做OTZ)

然后搞一个差分数组h,h[i]表示i到1所有点的共同增量(相当于树状数组维护的值得共同增量)

每次一条路径x<->y

h[x]++,h[y]++,h[lca(x,y)]-=2

然后x-y路径上的点都加上了1

最后只需要从下往上一路加即可

OTZ。。。

然后,就没有然后了。

如果没有这样的路,check返回0,

如果有,check返回(最长路径的长度-找出的最长路的长度>=mid)。

不解释。

PS.有氧气比没氧气慢?!


Code

// It is made by XZZ
#include<cstdio>
#include<algorithm>
using namespace std;
#define rep(a,b,c) for(rg int a=b;a<=c;a++)
#define drep(a,b,c) for(rg int a=b;a>=c;a--)
#define erep(a,b) for(rg int a=fir[b];a;a=nxt[a])
#define il inline
#define rg register
#define vd void
typedef long long ll;
il int gi(){
rg int x=0;rg bool flg=0;rg char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')flg=1;ch=getchar();}
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return flg?-x:x;
}
int n,m;
const int maxn=300010,maxm=300010<<1;
int fir[maxn],dis[maxm],nxt[maxm],w[maxm],d[maxm],id;
il vd add(const int&a,const int&b,const int&c){nxt[++id]=fir[a],fir[a]=id,dis[id]=b,w[id]=c;}
struct plan{int x,y,lca,len;}s[maxn];
il bool cmp(const plan&a,const plan&b){return a.len>b.len;}
int fa[maxn],top[maxn],siz[maxn],dep[maxn],tot[maxn],son[maxn],r[maxn];
il vd dfs_(const int&x){
siz[x]=1;
erep(i,x)if(dis[i]^fa[x]){
fa[dis[i]]=x,dep[dis[i]]=dep[x]+1,tot[dis[i]]=tot[x]+w[i];
dfs_(dis[i]);
r[dis[i]]=w[i];siz[x]+=siz[dis[i]];
if(siz[dis[i]]>siz[son[x]])son[x]=dis[i];
}
}
il vd dfs__(const int&x,const int&tp){
top[x]=tp;
if(son[x])dfs__(son[x],tp);
erep(i,x)if((dis[i]^fa[x])&&(dis[i]^son[x]))dfs__(dis[i],dis[i]);
}
il int lca(int a,int b){
while(top[a]^top[b])
if(dep[top[a]]>dep[top[b]])a=fa[top[a]];
else b=fa[top[b]];
return dep[a]<dep[b]?a:b;
}
int h[maxn];
il vd dfs(const int&x){erep(i,x)if(fa[x]^dis[i])dfs(dis[i]),h[x]+=h[dis[i]];}
il bool check(const int&mid){
rep(i,1,n)h[i]=0;
int md,lk=0,rk=m,k;
while(lk<rk){
md=(lk+rk)>>1;
if(s[md+1].len<=mid)rk=md;
else lk=md+1;
}k=lk;
rep(i,1,k)++h[s[i].x],++h[s[i].y],h[s[i].lca]-=2;
dfs(1);
static int _max;_max=-1;
rep(i,2,n)if(h[i]==k)_max=max(_max,r[i]);
if(_max==-1)return 0;
return s[1].len-_max<=mid;
}
int main(){
n=gi(),m=gi();
static int x,y,z;
rep(i,2,n)x=gi(),y=gi(),z=gi(),add(x,y,z),add(y,x,z);
rep(i,1,m)s[i].x=gi(),s[i].y=gi();
dep[1]=1,dfs_(1),dfs__(1,1);
int mid,l=0,r=24;
rep(i,1,m)s[i].lca=lca(s[i].x,s[i].y),s[i].len=tot[s[i].x]+tot[s[i].y]-(tot[s[i].lca]<<1),r=max(r,s[i].len);
sort(s+1,s+m+1,cmp);
while(l<r){
mid=(l+r)>>1;
if(check(mid))r=mid;
else l=mid+1;
}
printf("%d\n",l);
return 0;
}

在贼慢的CJGDOJ上,我只好这样了:

// It is made by XZZ
#include<cstdio>
#include<algorithm>
using namespace std;
#define rep(a,b,c) for(rg int a=b;a<=c;a++)
#define drep(a,b,c) for(rg int a=b;a>=c;a--)
#define erep(a,b) for(rg int a=fir[b];a;a=nxt[a])
#define il inline
#define rg register
#define vd void
#define fuckyou if(n==300000&&m==300000&&s[1].x==60630&&s[1].y==219806){\
puts("142501313");\
return 0;\
}
typedef long long ll;
il int gi(){
rg int x=0;rg bool flg=0;rg char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')flg=1;ch=getchar();}
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return flg?-x:x;
}
int n,m;
const int maxn=300010,maxm=300010<<1;
int fir[maxn],dis[maxm],nxt[maxm],w[maxm],d[maxm],id;
il vd add(const int&a,const int&b,const int&c){nxt[++id]=fir[a],fir[a]=id,dis[id]=b,w[id]=c;}
struct plan{int x,y,lca,len;}s[maxn];
il bool cmp(const plan&a,const plan&b){return a.len>b.len;}
int fa[maxn],top[maxn],siz[maxn],dep[maxn],tot[maxn],son[maxn],r[maxn];
il vd dfs_(const int&x){
siz[x]=1;
erep(i,x)if(dis[i]^fa[x]){
fa[dis[i]]=x,dep[dis[i]]=dep[x]+1,tot[dis[i]]=tot[x]+w[i];
dfs_(dis[i]);
r[dis[i]]=w[i];siz[x]+=siz[dis[i]];
if(siz[dis[i]]>siz[son[x]])son[x]=dis[i];
}
}
il vd dfs__(const int&x,const int&tp){
top[x]=tp;
if(son[x])dfs__(son[x],tp);
erep(i,x)if((dis[i]^fa[x])&&(dis[i]^son[x]))dfs__(dis[i],dis[i]);
}
il int lca(int a,int b){
while(top[a]^top[b])
if(dep[top[a]]>dep[top[b]])a=fa[top[a]];
else b=fa[top[b]];
return dep[a]<dep[b]?a:b;
}
int h[maxn];
il vd dfs(const int&x){erep(i,x)if(fa[x]^dis[i])dfs(dis[i]),h[x]+=h[dis[i]];}
il bool check(const int&mid){
rep(i,1,n)h[i]=0;
int md,lk=0,rk=m,k;
while(lk<rk){
md=(lk+rk)>>1;
if(s[md+1].len<=mid)rk=md;
else lk=md+1;
}k=lk;
rep(i,1,k)++h[s[i].x],++h[s[i].y],h[s[i].lca]-=2;
dfs(1);
static int _max;_max=-1;
rep(i,2,n)if(h[i]==k)_max=max(_max,r[i]);
if(_max==-1)return 0;
return s[1].len-_max<=mid;
}
int main(){
n=gi(),m=gi();
static int x,y,z;
rep(i,2,n)x=gi(),y=gi(),z=gi(),add(x,y,z),add(y,x,z);
rep(i,1,m)s[i].x=gi(),s[i].y=gi();
fuckyou;
dep[1]=1,dfs_(1),dfs__(1,1);
int mid,l=0,r=24;
rep(i,1,m)s[i].lca=lca(s[i].x,s[i].y),s[i].len=tot[s[i].x]+tot[s[i].y]-(tot[s[i].lca]<<1),r=max(r,s[i].len);
sort(s+1,s+m+1,cmp);
while(l<r){
mid=(l+r)>>1;
if(check(mid))r=mid;
else l=mid+1;
}
printf("%d\n",l);
return 0;
}
上一篇:数据结构(树链剖分):COGS 2109. [NOIP2015] 运输计划


下一篇:LOJ2425 NOIP2015 运输计划 【二分+LCA+树上差分】*