hdu_3804_Query on a tree(树链剖分)
2016-06-27 21:52
477 查看
题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=3804
题意:给你一棵树,然后给出树上边的价值,然后给出x,y,问从1到x的边上不超过y的最大值为多少
题解:这题如果直接写裸线段树来在线更新会T飞,正解是把边的值存下来,把询问也存下来,然后都按照价值从小到大排序,然后从最小的开始找,找之前把边的值插进线段树,这样就成了查询区间最大值了,很巧妙的方法。还有就是这题的内存卡的很紧,用前向星爆内存了,我没想通,用vector就过了。很尴尬
#include<cstdio>
#include<vector>
#include<algorithm>
#define F(i,a,b) for(int i=a;i<=b;++i)
#define root 1,n,1
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
using namespace std;
const int N=1e5+5;
vector<int>g
;
struct vs{
int val,u,v;
bool operator<(const vs &b)const{return val<b.val;}
}va
;
struct ser{
int x,k,id;
bool operator<(const ser &b)const{return k<b.k;}
}ask
;
int t,n,m,x,y,c,tot,ans
,sum[N*4],sz
,fa
,dep
,hs
,top
,tid
,idx;
//线段树部分
void update(int x,int k,int l,int r,int rt){
if(l==r){sum[rt]=k;return;}
int m=(l+r)>>1;
if(x<=m)update(x,k,ls);
else update(x,k,rs);
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
int query(int L,int R,int l,int r,int rt){
if(L<=l&&r<=R)return sum[rt];
int m=(l+r)>>1,maxs=0;
if(L<=m)maxs=max(query(L,R,ls),maxs);
if(m<R)maxs=max(query(L,R,rs),maxs);
return maxs;
}
//树链部分
void dfs1(int u,int pre){
sz[u]=1,fa[u]=pre,dep[u]=dep[pre]+1,hs[u]=0;
for(int i=0;i<g[u].size();i++){
int vv=g[u][i];
if(vv!=pre){
dfs1(vv,u);
if(sz[vv]>sz[hs[u]])hs[u]=vv;
sz[u]+=sz[vv];
}
}
}
void dfs2(int u,int tp){
tid[u]=++idx,top[u]=tp;
if(hs[u])dfs2(hs[u],tp);
for(int i=0;i<g[u].size();i++){
int vv=g[u][i];
if(vv!=fa[u]&&vv!=hs[u])dfs2(vv,vv);
}
}
int findmax(int x){
int fx=top[x],an=0;
while(fx!=1)an=max(query(tid[fx],tid[x],root),an),x=fa[fx],fx=top[x];
if(x==1)return an;
return max(query(2,tid[x],root),an);
}
int main(){
scanf("%d",&t);
while(t--){
scanf("%d",&n);
F(i,1,n)g[i].clear();
dep[0]=0,sz[0]=0;
F(i,0,4*N-1)sum[i]=0;
F(i,1,n-1)scanf("%d%d%d",&va[i].u,&va[i].v,&va[i].val),g[va[i].u].push_back(va[i].v),g[va[i].v].push_back(va[i].u);
dfs1(1,0),idx=0,dfs2(1,1);
scanf("%d",&m),tot=1;
F(i,1,m)scanf("%d%d",&ask[i].x,&ask[i].k),ask[i].id=i;
sort(va+1,va+n),sort(ask+1,ask+1+m);
F(i,1,m){
while(tot<n&&va[tot].val<=ask[i].k){
if(dep[va[tot].u]>dep[va[tot].v])c=va[tot].u;else c=va[tot].v;
update(tid[c],va[tot].val,root),tot++;
}
ans[ask[i].id]=findmax(ask[i].x);
}
F(i,1,m)printf("%d\n",ans[i]==0?-1:ans[i]);
}
return 0;
}
题意:给你一棵树,然后给出树上边的价值,然后给出x,y,问从1到x的边上不超过y的最大值为多少
题解:这题如果直接写裸线段树来在线更新会T飞,正解是把边的值存下来,把询问也存下来,然后都按照价值从小到大排序,然后从最小的开始找,找之前把边的值插进线段树,这样就成了查询区间最大值了,很巧妙的方法。还有就是这题的内存卡的很紧,用前向星爆内存了,我没想通,用vector就过了。很尴尬
#include<cstdio>
#include<vector>
#include<algorithm>
#define F(i,a,b) for(int i=a;i<=b;++i)
#define root 1,n,1
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
using namespace std;
const int N=1e5+5;
vector<int>g
;
struct vs{
int val,u,v;
bool operator<(const vs &b)const{return val<b.val;}
}va
;
struct ser{
int x,k,id;
bool operator<(const ser &b)const{return k<b.k;}
}ask
;
int t,n,m,x,y,c,tot,ans
,sum[N*4],sz
,fa
,dep
,hs
,top
,tid
,idx;
//线段树部分
void update(int x,int k,int l,int r,int rt){
if(l==r){sum[rt]=k;return;}
int m=(l+r)>>1;
if(x<=m)update(x,k,ls);
else update(x,k,rs);
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
int query(int L,int R,int l,int r,int rt){
if(L<=l&&r<=R)return sum[rt];
int m=(l+r)>>1,maxs=0;
if(L<=m)maxs=max(query(L,R,ls),maxs);
if(m<R)maxs=max(query(L,R,rs),maxs);
return maxs;
}
//树链部分
void dfs1(int u,int pre){
sz[u]=1,fa[u]=pre,dep[u]=dep[pre]+1,hs[u]=0;
for(int i=0;i<g[u].size();i++){
int vv=g[u][i];
if(vv!=pre){
dfs1(vv,u);
if(sz[vv]>sz[hs[u]])hs[u]=vv;
sz[u]+=sz[vv];
}
}
}
void dfs2(int u,int tp){
tid[u]=++idx,top[u]=tp;
if(hs[u])dfs2(hs[u],tp);
for(int i=0;i<g[u].size();i++){
int vv=g[u][i];
if(vv!=fa[u]&&vv!=hs[u])dfs2(vv,vv);
}
}
int findmax(int x){
int fx=top[x],an=0;
while(fx!=1)an=max(query(tid[fx],tid[x],root),an),x=fa[fx],fx=top[x];
if(x==1)return an;
return max(query(2,tid[x],root),an);
}
int main(){
scanf("%d",&t);
while(t--){
scanf("%d",&n);
F(i,1,n)g[i].clear();
dep[0]=0,sz[0]=0;
F(i,0,4*N-1)sum[i]=0;
F(i,1,n-1)scanf("%d%d%d",&va[i].u,&va[i].v,&va[i].val),g[va[i].u].push_back(va[i].v),g[va[i].v].push_back(va[i].u);
dfs1(1,0),idx=0,dfs2(1,1);
scanf("%d",&m),tot=1;
F(i,1,m)scanf("%d%d",&ask[i].x,&ask[i].k),ask[i].id=i;
sort(va+1,va+n),sort(ask+1,ask+1+m);
F(i,1,m){
while(tot<n&&va[tot].val<=ask[i].k){
if(dep[va[tot].u]>dep[va[tot].v])c=va[tot].u;else c=va[tot].v;
update(tid[c],va[tot].val,root),tot++;
}
ans[ask[i].id]=findmax(ask[i].x);
}
F(i,1,m)printf("%d\n",ans[i]==0?-1:ans[i]);
}
return 0;
}
相关文章推荐
- 【HDU 5366】The mook jong 详解
- 【HDU 2136】Largest prime factor 详细图解
- 【HDU 1568】Fibonacci 数学公式 详解
- HDU 1568
- HDU1290
- HDU1568(Fobonacci公式)
- HDU ACM Step 2.2.2 Joseph(约瑟夫环问题)
- HDU 1405
- HDU 1297
- hdu 1205
- hdu 2087
- hdu 1016
- HDU 4898 The Revenge of the Princess’ Knight ( 2014 Multi-University Training Contest 4 )
- HDU 5592 ZYB's Premutation 线段树(查找动态区间第K大)
- HDU 5240 Exam (好水的题)
- HDU5237 Base64 大模拟
- HDU 1000
- HDU 1001
- HDU 1016 Prime Ring Problem
- HDU 1017 A Mathematical Curiosity