天天看点

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