hdu 4729 An Easy Problem for Elfness,函数式线段树
13年成都网赛的题。
主要是问一个树上的某两个点间的路径,你可以对某些边增加一些流量,最多这个路径上能流过多少流量。
题目意思比较复杂(废话多),最大流量可以分成这几种情况:
1、增加一条边,剩下的费用用来增加这条边的流量,加上本来两点间的流量。
2、所有的费用用来建新边,加上本来的流量。
3、在本来的边上增加流量,求最大能通过流量。
最近怎么老根各种树过不去。
函数式线段树能够处理某些树上路径问题。
简单来说就是这样,以边权为下标建线段树。
想象每个点都有一个[1..cmax]的线段树,而且,每个点的树是从父节点的树插入一个之间的边的流量的点得到。
这样呢,我们可以用u、v、lca[u,v]的线段树来表示u到v路径的线段树。
u、v路径的线段树就是t[u]+t[v]-2*t[lca]。
现在问题就是找到最大的c,使得费用足够使得小于c的边都变成c。
线段树是个好东西,O(logn)可以搞定。
需要记录两个值,总权值v,总边数cnt。
如果(总共能添加的边流量val>=左区间的上界m*总边数cnt-总权值v),那就向右边递归,否则向左边。
具体看代码吧:
13年成都网赛的题。
主要是问一个树上的某两个点间的路径,你可以对某些边增加一些流量,最多这个路径上能流过多少流量。
题目意思比较复杂(废话多),最大流量可以分成这几种情况:
1、增加一条边,剩下的费用用来增加这条边的流量,加上本来两点间的流量。
2、所有的费用用来建新边,加上本来的流量。
3、在本来的边上增加流量,求最大能通过流量。
最近怎么老根各种树过不去。
函数式线段树能够处理某些树上路径问题。
简单来说就是这样,以边权为下标建线段树。
想象每个点都有一个[1..cmax]的线段树,而且,每个点的树是从父节点的树插入一个之间的边的流量的点得到。
这样呢,我们可以用u、v、lca[u,v]的线段树来表示u到v路径的线段树。
u、v路径的线段树就是t[u]+t[v]-2*t[lca]。
现在问题就是找到最大的c,使得费用足够使得小于c的边都变成c。
线段树是个好东西,O(logn)可以搞定。
需要记录两个值,总权值v,总边数cnt。
如果(总共能添加的边流量val>=左区间的上界m*总边数cnt-总权值v),那就向右边递归,否则向左边。
具体看代码吧:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define NN 100010
struct que{
int s,t,k,a,b,lca;
void init(int y,int u,int i,int o,int p){
s=y;t=u;k=i;a=o;b=p;
}
}q[NN];
vector<pair<int,int> > fi[NN];
void addedge(int fr,int to,int val){
fi[fr].push_back(make_pair(to,val));
}
int root[NN],tott;
struct segtree{
int ls,rs,l,r,v,cnt;
void init(int val=0){ls=rs=l=r=0;v=val;cnt=0;}
}t[NN*20];
void init_segtree(){
tott=0;
t[0].init();
}
int build(int l,int r){
int p=++tott;
t[p].l=l;t[p].r=r;t[p].v=t[p].cnt=0;
if (l==r){
t[p].ls=t[p].rs=0;return p;
}
int m=(l+r)>>1;
t[p].ls=build(l,m);
t[p].rs=build(m+1,r);
return p;
}
int insert(int pos,int num,int pas){
int p=++tott;
t[p]=t[pas];
t[p].v+=pos;
t[p].cnt+=num;
if (t[p].l==t[p].r) return p;
int m=(t[p].l+t[p].r)>>1;
if (pos<=m) t[p].ls=insert(pos,num,t[p].ls);
else t[p].rs=insert(pos,num,t[p].rs);
return p;
}
int query1(int s,int to,int lca,int tc=0,int tv=0){
return t[s].r*(tc+t[s].cnt+t[to].cnt-t[lca].cnt-t[lca].cnt)-(tv+t[s].v+t[to].v-t[lca].v-t[lca].v);
}
int queryc(int s,int to,int lca){
return t[s].cnt+t[to].cnt-t[lca].cnt-t[lca].cnt;
}
int queryv(int s,int to,int lca){
return t[s].v+t[to].v-t[lca].v-t[lca].v;
}
int query(int s,int to,int lca,int val,int tc,int tv){
if (t[s].l==t[s].r){
if (query1(s,to,lca,tc,tv)<=val) return t[s].l;
else return t[s].l-1;
}
int tmp,m=t[s].l+t[s].r>>1;
if ((tmp=query1(t[s].ls,t[to].ls,t[lca].ls,tc,tv))<=val){
return query(t[s].rs,t[to].rs,t[lca].rs,val,tc+queryc(t[s].ls,t[to].ls,t[lca].ls),tv+queryv(t[s].ls,t[to].ls,t[lca].ls));
}
else {
return query(t[s].ls,t[to].ls,t[lca].ls,val,tc,tv);
}
}
void dfs(int u,int fa,int val){
int i,v,sz;
if (u!=1){
root[u]=insert(val,1,root[fa]);
}
sz=fi[u].size();
for(i=0;i<sz;++i){
v=fi[u][i].first;
if (v!=fa){
dfs(v,u,fi[u][i].second);
}
}
}
vector<pair<int,int> > vec[NN];
int set[NN],vis[NN];
int dep[NN];
int findset(int x){return x==set[x]?x:set[x]=findset(set[x]);}
void init_lca(int n){
for(int i=1;i<=n;++i){vec[i].clear();set[i]=i;vis[i]=0;}
}
void tarjan(int u,int fa,int de){
int i;
dep[u]=de;
int v,sz=fi[u].size();
for(i=0;i<sz;++i){
v=fi[u][i].first;
if (v!=fa){
tarjan(v,u,de+1);
set[v]=u;
}
}
vis[u]=1;
sz=vec[u].size();
for(i=0;i<sz;++i){
v=vec[u][i].first;
if (vis[v]) q[vec[u][i].second].lca=findset(v);
}
}
int main(){
//freopen("4729in.txt","r",stdin);
int tcas,cas,n,m,cmax,a,b,c,s,t,k;
int i,val,tp,tmp,ans,lca;
scanf("%d",&tcas);
for(cas=1;cas<=tcas;++cas){
scanf("%d%d",&n,&m);
for(i=1;i<=n;++i) fi[i].clear();
cmax=0;
for(i=1;i<n;++i){
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
if (c>cmax) cmax=c;
}
init_lca(n);
for(i=1;i<=m;++i){
scanf("%d%d%d%d%d",&s,&t,&k,&a,&b);
q[i].init(s,t,k,a,b);
vec[s].push_back(make_pair(t,i));
vec[t].push_back(make_pair(s,i));
}
tarjan(1,-1,0);
init_segtree();
root[1]=build(1,cmax);
dfs(1,-1,0);
printf("Case #%d:\n",cas);
for(i=1;i<=m;++i){
a=q[i].a;b=q[i].b;k=q[i].k;s=q[i].s;t=q[i].t;lca=q[i].lca;
tmp=query(root[s],root[t],root[lca],0,0,0);
if (k>=a) ans=max(k/a+tmp,(k-a)/b+1+tmp);
else ans=0;
val=k/b;
tp=dep[s]+dep[t]-dep[lca]*2;
tmp=query1(root[s],root[t],root[lca]);
if (tmp<=val){
ans=max(ans,cmax+(val-tmp)/tp);
}
else {
ans=max(query(root[s],root[t],root[lca],val,0,0),ans);
}
printf("%d\n",ans);
}
}
return 0;
}