1. 程式人生 > >動態dp模板

動態dp模板

noip居然考了這個毒瘤玩意,雖然有不用ddp的方法,但是還是簡單學一下,QAQ

對於dp的轉移可以用矩陣乘法的方式來維護,只不過修改了一些矩陣乘法中的運算型別,本質還是一樣的,由於矩陣有結合律,所以可以用個線段樹等資料結構維護,對於樹上的,我們用dfs序或者樹鏈剖分就可以轉化為序列上的問題。

對於樹上的用樹剖的話是 O ( (

) × n l o
g 2 n ) O((矩陣乘法複雜度)\times nlog^2n) 的,所以我們有時可以用 L C
T LCT
來優化到 O ( ( ) × n l o g n ) O((矩陣乘法複雜度)\times nlogn) ,但是 L C T LCT 常數巨大,不一定有樹剖跑的快,所以還可以用全域性平衡二叉樹優化,這裡不再多說。

樹剖模板【IN-Luogu】(過不去加強版的,要一個 l o g n logn 的才過的去)

#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;
}


準備練習相關題目:


乘風破浪會有時,直掛雲帆濟滄海

加油努力!