動態dp模板
阿新 • • 發佈:2018-12-19
noip居然考了這個毒瘤玩意,雖然有不用ddp的方法,但是還是簡單學一下,QAQ
對於dp的轉移可以用矩陣乘法的方式來維護,只不過修改了一些矩陣乘法中的運算型別,本質還是一樣的,由於矩陣有結合律,所以可以用個線段樹等資料結構維護,對於樹上的,我們用dfs序或者樹鏈剖分就可以轉化為序列上的問題。
對於樹上的用樹剖的話是 的,所以我們有時可以用 來優化到 ,但是 常數巨大,不一定有樹剖跑的快,所以還可以用全域性平衡二叉樹優化,這裡不再多說。
樹剖模板【IN-Luogu】(過不去加強版的,要一個 的才過的去)
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll int
using namespace std;
const int N=1e5+10;
const int M=1e5+15;
int n,m;ll P[N];
struct ss{
int to,last;
ss(){}
ss(int a,int b):to(a),last(b){}
}g[N<<1];
int head[N],cnt;
void add(int a,int b){
g[++cnt]=ss(b,head[a]);head[a]=cnt;
g[++cnt]=ss(a,head[b]);head[b]=cnt;
}
int pos[M],rf[M],top[M],f[M],son[M],sze[M],ed[M];
int tim;
void dfs1(int a){
sze[a]=1;
for(int i=head[a];i;i=g[i].last){
if(g[i].to==f[a]) continue;
f[g[i].to]=a;
dfs1(g[i].to);
sze[a]+=sze[g[i].to];
if(!son[a]||sze[son[a]]<sze[g[i].to])
son[a]=g[i].to;
}
}
void dfs2(int a,int b){
top[a]=b;rf[pos[a]=++tim]=a;
if(!son[a]){
ed[b]=tim;
return;
}
dfs2(son[a],b);
for(int i=head[a];i;i=g[i].last){
if(g[i].to==f[a]||g[i].to==son[a]) continue;
dfs2(g[i].to,g[i].to);
}
}
ll dp[2][M];
void init(int a){
dp[0][a]=0;
dp[1][a]=max(0,P[a]);
for(int i=head[a];i;i=g[i].last){
if(g[i].to==f[a]) continue;
init(g[i].to);
dp[0][a]+=max(dp[0][g[i].to],dp[1][g[i].to]);
dp[1][a]+=dp[0][g[i].to];
}
}
struct matrix{
ll v[2][2];
void clear(){memset(v,0,sizeof(v));}
}ls;
matrix operator *(const matrix &a,const matrix &b){
ls.v[0][0]=max(a.v[0][0]+b.v[0][0],a.v[0][1]+b.v[1][0]);
ls.v[0][1]=max(a.v[0][0]+b.v[0][1],a.v[0][1]+b.v[1][1]);
ls.v[1][0]=max(a.v[1][0]+b.v[0][0],a.v[1][1]+b.v[1][0]);
ls.v[1][1]=max(a.v[1][0]+b.v[0][1],a.v[1][1]+b.v[1][1]);
return ls;
}
/*
matrix operator *(const matrix &a,const matrix &b){
matrix ls;
for(int i=0;i<=1;i++){
for(int j=0;j<=1;j++){
ls.v[i][j]=0;
for(int k=0;k<=1;k++){
ls.v[i][j]=max(ls.v[i][j],a.v[i][k]+b.v[k][j]);
}
}
}
return ls;
}
*/
struct Segment_tree{
matrix S[M<<2],rec[M];
void pushup(int o){
S[o]=S[o<<1]*S[o<<1|1];
}
void build(int o,int l,int r){
if(l==r){
int a=rf[l];
ll f0=0,f1=P[a];
for(int i=head[a];i;i=g[i].last){
if(g[i].to==f[a]||g[i].to==son[a]) continue;
f0+=max(dp[0][g[i].to],dp[1][g[i].to]);
f1+=dp[0][g[i].to];
}
S[o].v[0][0]=S[o].v[0][1]=f0;
S[o].v[1][0]=f1;S[o].v[1][1]=0;
rec[l]=S[o];
return;
}
int mid=l+r>>1;
build(o<<1,l,mid);
build(o<<1|1,mid+1,r);
pushup(o);
}
void update(int o,int l,int r,int p){
if(l==r){
S[o]=rec[l];
return;
}
int mid=l+r>>1;
if(p<=mid) update(o<<1,l,mid,p);
else update(o<<1|1,mid+1,r,p);
pushup(o);
}
matrix query(int o,int l,int r,int L,int R){
if(L<=l&&r<=R) return S[o];
int mid=l+r>>1;
if(R<=mid) return query(o<<1,l,mid,L,R);
else if(L>mid) return query(o<<1|1,mid+1,r,L,R);
else return query(o<<1,l,mid,L,R)*query(o<<1|1,mid+1,r,L,R);
}
matrix query_chain(int o){
return query(1,1,n,pos[o],ed[o]);
}
void calc(int p,ll tp){
int id=pos[p];
rec[id].v[1][0]+=tp-P[p];
P[p]=tp;
matrix bef,now;
while(p){
bef=query_chain(top[p]);
update(1,1,n,pos[p]);
now=query_chain(top[p]);
p=f[top[p]];
if(!p) break;
id=pos[p];
rec[id].v[0][0]+=max(now.v[0][0],now.v[1][0])-max(bef.v[0][0],bef.v[1][0]);
rec[id].v[0][1]=rec[id].v[0][0];
rec[id].v[1][0]+=now.v[0][0]-bef.v[0][0];
}
}
}Seg;
void solve(int p,ll t){
Seg.calc(p,t);
matrix now=Seg.query_chain(1);
printf("%d\n",max(now.v[0][0],now.v[1][0]));
}
void Init_bef(){
dfs1(1);
dfs2(1,1);
init(1);
Seg.build(1,1,n);
}
int a,b;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&P[i]);
}
for(int i=1;i<n;i++){
scanf("%d%d",&a,&b);
add(a,b);
}
Init_bef();
while(m--){
scanf("%d%d",&a,&b);
solve(a,b);
}
return 0;
}
準備練習相關題目:
- SDOI2017切樹遊戲BlogIN
- [BZOJ5210]-最大連通子塊和
- 【#268. 【清華集訓2016】資料互動】BlogIN
乘風破浪會有時,直掛雲帆濟滄海
加油努力!