[JZOJ5976] 打怪兽 【DP】【决策单调性】

Description

[JZOJ5976] 打怪兽 【DP】【决策单调性】_第1张图片
[JZOJ5976] 打怪兽 【DP】【决策单调性】_第2张图片

Solution

首先发现性质

可以看出,如果上一次叠的甲还没有掉完,那么此时是不会叠甲的
因为这时候叠甲,不如把这些甲移到上次一起叠,那么肯定是更优的。

那么现在就相当于用若干个下降且不交的三角形来覆盖这个序列。

考虑DP
F [ i ] [ j ] F[i][j] F[i][j]表示1到i-1我们已经处理完了,已经花了j层甲,现在是一层也没有的最大总挡掉的伤害值。
要么直接从i-1不叠甲转移过来,要么从前面某一个k的位置叠i-k层甲,到i处影响刚好结束。

那么预处理c[i][j]表示在i处叠j甲总共能挡掉多少伤害,特殊处理不叠的情况,转移就是 f [ i ] [ j ] = m a x ( f [ k ] [ j − i + k ] + c [ k ] [ i − k ] ) , k ∈ [ 1 , i − 1 ] f[i][j]=max(f[k][j-i+k]+c[k][i-k]),k\in[1,i-1] f[i][j]=max(f[k][ji+k]+c[k][ik]),k[1,i1]
如果我们令 p = i − k p=i-k p=ik,则 f [ i ] [ j ] = m a x ( f [ i − p ] [ j − p ] + c [ i − p ] [ p ] ) f[i][j]=max(f[i-p][j-p]+c[i-p][p]) f[i][j]=max(f[ip][jp]+c[ip][p])
我们发现,在这个过程中能转移的状态 i − j i-j ij是恒定的

那么把所有状态按i-j分组,令 d = i − j d=i-j d=ij
那么状态就变成 f [ i ] [ i − d ] f[i][i-d] f[i][id]

对于一个 f [ i ] [ i − d ] f[i][i-d] f[i][id],考虑能转移到它的两个位置 x , y x,y x,y,我们假设x 那么 f [ x ] [ x − d ] + c [ x ] [ i − x ] > f [ y ] [ y − d ] + c [ y ] [ i − y ] f[x][x-d]+c[x][i-x]>f[y][y-d]+c[y][i-y] f[x][xd]+c[x][ix]>f[y][yd]+c[y][iy]
考虑 c [ x ] [ i − x ] − c [ y ] [ i − y ] c[x][i-x]-c[y][i-y] c[x][ix]c[y][iy],它是从x开始的三角形减去从y开始的三角形的部分
大概如下图

[JZOJ5976] 打怪兽 【DP】【决策单调性】_第3张图片
红色部分挡掉的伤害就是 c [ x ] [ i − x ] − c [ y ] [ i − y ] c[x][i-x]-c[y][i-y] c[x][ix]c[y][iy],可以发现它随着i的增大是单调递增的,而 f [ x ] [ x − d ] , f [ y ] [ y − d ] f[x][x-d],f[y][y-d] f[x][xd],f[y][yd]与i无关

因此我们可以得出,这个决策是单调的,位置更前的决策x,在i越后的时候就会越优,直到在某个位置超过y

这就是典型的1D1D动态规划用决策单调性的优化了
那么对于每一组,我们可以用一个单调栈来存储决策,每个元素记录它被它下面的元素超过的时间,i右移、新加入元素的时候相应的弹栈即可。

此时我们还需要查询两个位置超过的时间,这可以用二分来做。

因此总的复杂度是 O ( n 2 log ⁡ n ) 的 O(n^2\log n)的 O(n2logn)

Code

#include 
#include 
#include 
#include 
#include 
#include 
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fod(i,a,b) for(int i=a;i>=b;--i)
#define N 4005
#define LL long long
using namespace std;
LL f[N][N],c[N][2*N],ans[N];
int n,a[N],d[N][N],t[N][N],le[N];
int fd(int q,int y,int x)
{
	int l=y+1,r=2*n+1;
	while(l+1<r)
	{
		int mid=(l+r)>>1;
		if(f[y][y-q]+c[y][mid-y]<f[x][x-q]+c[x][mid-x]) r=mid;
		else l=mid;
	}
	if(f[y][y-q]+c[y][l-y]<f[x][x-q]+c[x][l-x]) return l;
	else return r;
}
void push(int q,int k)
{
	while(le[q]>0&&t[q][le[q]]<=fd(q,k,d[q][le[q]])) d[q][le[q]]=t[q][le[q]]=0,le[q]--;
	d[q][++le[q]]=k;
	if(le[q]==1) t[q][le[q]]=2*n+1;
	else t[q][le[q]]=fd(q,k,d[q][le[q]-1]);
}
LL get(int q,int k)
{
	if(le[q]==0) return 0;
	while(le[q]>0&&t[q][le[q]]<=k) d[q][le[q]]=t[q][le[q]]=0,le[q]--;
	return f[d[q][le[q]]][d[q][le[q]]-q]+c[d[q][le[q]]][k-d[q][le[q]]];
}
int main()
{
	cin>>n;
	LL s=0;
	fo(i,1,n) scanf("%d",&a[i]),s+=a[i];
	fod(i,n,1)
	{
		fo(j,1,2*n) 
		{
			c[i][j]=min(j,a[i])+c[i+1][j-1];
			c[i][j]=max(c[i][j],c[i][j-1]);
		}
	}
	push(1,1);
	push(0,1);
	fo(i,2,2*n)
	{
		fo(j,0,i)
		{
			if(j>n) break;
			if(i==n+1) ans[j]=max(ans[j],f[i-1][j]);
			else if(i<=n) f[i][j]=max(f[i][j],f[i-1][j]);
			if(i>n) ans[j]=max(ans[j],get(i-j,i));
			else 
			{
				f[i][j]=max(f[i][j],get(i-j,i));
				push(i-j,i);
			}
		}
	}
	fo(i,1,n) printf("%lld\n",s-ans[i]);
}

你可能感兴趣的:(题解,————其他dp,————单调栈,---DP,————其他DP优化)