light oj 1128 倍增法LCA + DP

來源:互聯網
上載者:User

題意:一棵樹,每個點都有一個點權,給你q個詢問a b,求a到根節點的路徑中點權大於等於b且最接近根節點的點編號


剛開始想用樹鏈剖分搞,後來聽說用倍增+DP也行

做法:

如果懂倍增法求祖先的方法的話,那麼這個DP就能很快理解了,詳情見代碼

dp[i][j]表示i到i的2^j祖先的路徑上的點權的最大值

那麼在找的時候只需要log(n)的複雜度就可以找到最上面的點權大於等於b的啦

#include<cstdio>#include<cstring>#include<vector>#include<algorithm>using namespace std;const int POW = 18;const int maxn = 100010;vector<int> edge[maxn];int val[maxn];int f[maxn][POW];int dp[maxn][POW];void dfs(int u){int sz=edge[u].size();for(int i=0;i<sz;i++){int v=edge[u][i];f[v][0]=u;dp[v][0]=val[v];for(int j=1;j<POW;j++) f[v][j]=f[f[v][j-1]][j-1];for(int j=1;j<POW;j++) dp[v][j]=max(dp[v][j-1],dp[f[v][j-1]][j-1]);dfs(v);}}int solve(int s,int v){for(int i=POW-1;i>=0;i--){if(dp[f[s][i]][i]>=v)s=f[s][i];}return s;}int main(){int t,ca=1,n,q;scanf("%d",&t);while(t--){scanf("%d%d",&n,&q);for(int i=0;i<n;i++) edge[i].clear();val[0]=1;for(int i=0;i<POW;i++)dp[0][i]=val[0],f[0][i]=0;for(int i=1,fa;i<n;i++){scanf("%d%d",&fa,&val[i]);edge[fa].push_back(i);}dfs(0);int s,v;printf("Case %d:\n",ca++);while(q--){scanf("%d%d",&s,&v);printf("%d\n",solve(s,v));}}return 0;}

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.