題目鏈接：點擊查看

題目大意：給出一棵樹，每條邊上都有一個權值，給出m個查詢：a，b：問從點1到點a的唯一路徑上，在邊權小于等于b的邊中選出邊權最大的值輸出，若沒有符合條件的邊則輸出-1；

題目分析：時隔三個月再次重溫這個題目，感覺這個題目真的就簡單多了，直接剖完然后離線線段樹做就行了，所以為了挑戰一下自己，我選擇用主席樹寫了一發，不得不說，這個題目內存卡的真死，權值的數據給到了1e9，本來想開32倍的主席樹然后直接用線段樹動態離散化的，結果第一發直接MLE了，看了一眼題目給的內存，只給了32M，開32倍的線段樹就直接38M了，無奈，只能離散化一下了，還好這個題沒卡我用vector離散化，最后31.7M險些飄過

這里和zx學長在討論這個題的時候有點對主席樹的小理解吧，我之前有點想當然的以為主席樹都應該維護一個sum來判斷是否有滿足條件的區間，但其實并不然，就拿這個題為例，因為是要求點1到點u這條鏈上滿足條件的最大值，所以我們只用維護一個最大值的主席樹即可，并不用維護sum這個變量了，zx學長給我這么一點撥，我就想到若想求u-v這條邊上滿足條件的最大值的話，是不是就不太好用主席樹來做了，zx學長只是給我留了一句話，可以用主席樹做，只不過時間復雜度會帶上個常數，emmm，說實話我想了很久也沒想到該怎么實現，可能是我還太菜了，我現在對主席樹的理解還是只會維護一個sum和，然后用sum解決第k大的問題以及大于等于或者小于等于k的數有多少個，要是真的讓寫從u-v上滿足條件的最大值的話，我還是只會卑微的用線段樹離線來寫

希望過三個月再回顧現在寫的這些內容，會感覺自己之前的理解都好幼稚，也希望能看到進步之后的自己，就像現在的自己看三個月之前的博客一樣吧

代碼：

#include<iostream> #include<cstdlib> #include<string> #include<cstring> #include<cstdio> #include<algorithm> #include<climits> #include<cmath> #include<cctype> #include<stack> #include<queue> #include<list> #include<vector> #include<set> #include<map> #include<sstream> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=1e5+100;int n;struct Node {int to,w,next; }edge[N<<1];int head[N],tot;void addedge(int u,int v,int w)//鏈式前向星 {edge[tot].to=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;edge[tot].to=u;edge[tot].w=w;edge[tot].next=head[v];head[v]=tot++; }vector<int>num;//離散化蛹int getnum(int x)//離散化用 {return lower_bound(num.begin(),num.end(),x)-num.begin()+1; }struct tree {int l,r,mmax; }tree[N*32];//主席樹維護最大值int root[N],cnt;void insert(int pre,int& cur,int l,int r,int pos) {cur=++cnt;tree[cur]=tree[pre];tree[cur].mmax=max(tree[pre].mmax,pos);if(l==r)return;int mid=l+r>>1;if(mid>=pos)insert(tree[pre].l,tree[cur].l,l,mid,pos);elseinsert(tree[pre].r,tree[cur].r,mid+1,r,pos); }int query(int cur,int l,int r,int ll,int rr)//l,r:當前區間 ll,rr:目標區間 {if(l>=ll&&r<=rr)return tree[cur].mmax;if(r<ll||l>rr)return 0;int mid=l+r>>1;return max(query(tree[cur].l,l,mid,ll,rr),query(tree[cur].r,mid+1,r,ll,rr)); }void dfs(int u,int f) {for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].to;int w=edge[i].w;if(v==f)continue;insert(root[u],root[v],1,num.size(),getnum(w));dfs(v,u);} }void init() {for(int i=1;i<=n;i++)head[i]=-1;cnt=tot=0; }int main() { // freopen("input.txt","r",stdin); // ios::sync_with_stdio(false);int w;cin>>w;while(w--){scanf("%d",&n);init();num.clear();for(int i=1;i<n;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);addedge(u,v,w);num.push_back(w);}sort(num.begin(),num.end());num.erase(unique(num.begin(),num.end()),num.end());dfs(1,0);int m;scanf("%d",&m);while(m--){int x,y;scanf("%d%d",&x,&y);int pos=upper_bound(num.begin(),num.end(),y)-num.begin();int ans=query(root[x],1,num.size(),1,pos);if(ans)printf("%d\n",num[ans-1]);elseprintf("%d\n",-1);}}return 0; }

?

總結

以上是生活随笔為你收集整理的HDU - 3804 Query on a tree(主席树维护最大值+离散化)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。