天天看點

BZOJ 4621 Tc605 (DP)

題目描述:http://www.lydsy.com/JudgeOnline/problem.php?id=4621

題解:

觀察到原排列進行k次操作後同種數字必成為連續的一段,我們可以定義f[i,j]表示前i個數字中進行了j次操作。

DP時先暴力找出可以把目前a[i]擴充到的區間找到,記為[l, r]。那麼可以得出一個非常顯然的DP方程:

dp[i][j]+=Σdp[k][j-1], l-1<=k<r (具體情況畫一下圖應該比較就能得出來了)

特别的,這樣計算時會将本次操作區間長度為1的算進去,即有:f[i][j]+=f[i-1][j-1]

但這樣的操作是沒有意義的,對原數列沒有一點改變,是以要将這一步的影響消除:

f[i][j-1]=f[i][j-1]+f[i-1][j-1];f[i][j]=f[i][j]-f[i-1][j-1];(不計算這一步操作)

時間複雜度為O(n^3)

參考代碼:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
#define rep(i,j,k) for(int i=(int)j;i<=(int)k;i++)
#define per(i,j,k) for(int i=(int)j;i>=(int)k;i--)
#define mod 1000000007

int n,m,a[510],f[510][510];
int main(){
	scanf("%d%d",&n,&m);
	rep(i,1,n)scanf("%d",&a[i]);
	f[0][0]=1;
	int l,r,tmp;
	rep(i,1,n){
		l=r=i;
		while(l>1 && a[l-1]<a[i])l--;
		while(r<n && a[r+1]<a[i])r++;
		f[i][m]=(f[i][m]+f[i-1][m])%mod;
		per(j,m,1){
			tmp=f[l-1][j-1];
			rep(k,l,r){
				f[k][j]=(f[k][j]+tmp)%mod;
				tmp=(tmp+f[k][j-1])%mod;
			}
			f[i][j-1]=(f[i][j-1]+f[i-1][j-1])%mod;
			f[i][j]=(f[i][j]-f[i-1][j-1]+mod)%mod;
		}
	}
	int ans;
	rep(i,0,m)ans=(ans+f[n][i])%mod;
	printf("%d\n",ans);
	return 0;
}