[Bzoj4182]Shopping(点分治)(树上背包)(单调队列优化多重背包)
生活随笔
收集整理的這篇文章主要介紹了
[Bzoj4182]Shopping(点分治)(树上背包)(单调队列优化多重背包)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
4182: Shopping
?
Time Limit:?30 Sec??Memory Limit:?128 MBSubmit:?374??Solved:?130
[Submit][Status][Discuss]
Description
?
馬上就是小苗的生日了,為了給小苗準備禮物,小蔥興沖沖地來到了商店街。商店街有n個商店,并且它們之間的道路構成了一顆樹的形狀。
第i個商店只賣第i種物品,小苗對于這種物品的喜愛度是wi,物品的價格為ci,物品的庫存是di。但是商店街有一項奇怪的規定:如果在商店u,v買了東西,并且有一個商店w在u到v的路徑上,那么必須要在商店w買東西。小蔥身上有m元錢,他想要盡量讓小苗開心,所以他希望最大化小苗對買 到物品的喜愛度之和。這種小問題對于小蔥來說當然不在話下,但是他的身邊沒有電腦,于是他打電話給同為OI選手的你,你能幫幫他嗎?Input
?
輸入第一行一個正整數T,表示測試數據組數。
對于每組數據, 第一行兩個正整數n;m; 第二行n個非負整數w1,w2...wn; 第三行n個正整數c1,c2...cn; 第四行n個正整數d1,d2...dn; 接下來n-1行每行兩個正整數u;v表示u和v之間有一條道路Output
?
輸出共T 行,每行一個整數,表示最大的喜愛度之和。Sample Input
?
1 3 2 1 2 3 1 1 1 1 2 1 1 2 1 3?
Sample Output
?
4?
HINT
?
?
?N<=500,M<=4000,T<=5,Wi<=4000,Di<=100
?
?
分析:
題意:兩個點選了,它路徑上的點必須選。求樹上一個聯通塊的多重背包,權值最大。
?
題解: 先用點分治假設重心必選,然后dfs它子樹,這樣每個點會做背包的次數降低到log次。 然后dfs子樹時列出dfs序,然后轉移方程:?
對于第二步,多重背包優化可以考慮二進制拆分總復雜度為O(Tnmlognlogm)。
也可以使用單調隊列優化總復雜度O(Tnmlogn)
下面是對單調隊列優化的圖解:
嗯。。沒了。
AC代碼:
?
# include <cstdio> # include <cstring> # include <iostream> # include <algorithm> using namespace std; const int N = 4e3 + 12; const int M = 5e3 + 12; int mx[N],w[N],v[N],c[N],n,m,root,head[N],dt,sz[N],sum,id[N],ed[N],ans; int f[N][M]; bool vis[N]; struct Edge{int to,nex; }edge[N << 1]; void AddEdge(int u,int v) {edge[++dt] = (Edge){v,head[u]};head[u] = dt; } void find(int u,int pre) {mx[u] = 0;sz[u] = 1;for(int i = head[u];i;i = edge[i].nex){if(vis[edge[i].to] || edge[i].to == pre)continue;find(edge[i].to,u);sz[u] += sz[edge[i].to];mx[u] = max(mx[u],sz[edge[i].to]);}mx[u] = max(mx[u],sum - sz[u]);if(mx[u] < mx[root])root = u; } void dfs(int u,int pre) {sz[u] = 1;id[++dt] = u;for(int i = head[u];i;i = edge[i].nex){if(vis[edge[i].to] || edge[i].to == pre)continue;dfs(edge[i].to,u);sz[u] += sz[edge[i].to];}ed[u] = dt; } int Q1[M],Q2[M]; void calc(int *g,int x) {int h1,h2,t1,t2,cnt,t;for(int j = 0;j < v[x];j++){h1 = t1 = h2 = t2 = cnt = 0;for(int k = j;k <= m;k += v[x]){if(t1 - h1 == c[x] + 1){if(Q2[h2 + 1] == Q1[h1 + 1])++h2;++h1;}t = g[k] - cnt * w[x];Q1[++t1] = t;while(h2 < t2 && Q2[t2] < t)--t2;Q2[++t2] = t;g[k] = Q2[h2 + 1] + cnt * w[x];++cnt;}} } void solve() {for(int i = 1;i <= dt + 1;i++)for(int j = 0;j <= m;j++)f[i][j] = 0;int x,t;for(int i = dt;i >= 1;i--){x = id[i]; for(int j = m;j >= v[x];j--)f[i][j] = f[i + 1][j - v[x]] + w[x];calc(f[i],x);for(int j = m;j >= 0;j--)f[i][j] = max(f[i][j],f[ed[x] + 1][j]);}ans = max(ans,f[1][m]); } void dfs(int u) {vis[u] = true;dt = 0;dfs(u,-1);solve();for(int i = head[u];i;i = edge[i].nex){if(vis[edge[i].to])continue;root = 0;sum = sz[edge[i].to];if(sum > sz[u])sum = sum - sz[u];find(edge[i].to,u);dfs(root);} } int main() {mx[0] = N;int Case;scanf("%d",&Case);while(Case--){scanf("%d %d",&n,&m);int x,y;memset(vis,false,sizeof vis);memset(head,0,sizeof head);dt = ans = 0;for(int i = 1;i <= n;i++)scanf("%d",&w[i]);for(int i = 1;i <= n;i++)scanf("%d",&v[i]);for(int i = 1;i <= n;i++)scanf("%d",&c[i]),c[i]--;for(int i = 1;i < n;i++)scanf("%d %d",&x,&y),AddEdge(x,y),AddEdge(y,x);root = 0;sum = n;find(1,-1);dfs(root);printf("%d\n",ans);} }?
?
?
?
轉載于:https://www.cnblogs.com/lzdhydzzh/p/8672725.html
總結
以上是生活随笔為你收集整理的[Bzoj4182]Shopping(点分治)(树上背包)(单调队列优化多重背包)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 怀孕梦到猫咬自己是什么意思
- 下一篇: 人梦到别人敲木鱼代表什么