YbtOJ 单调队列课堂过关 例2 粉刷木板【单调队列】【DP】

题目

思路

设 $f_{i,j}$ 表示前 $i$ 个粉刷匠，刷了前 $j$ 个木块（可以有木块不选）
然后转移不刷 $$f_{i,j}=max(f_{i,j-1},f_{i-1,j})$$
和刷 $$\underset{j-l_i\le k<s_i}{\max }{f}_{i-1,k}+p_i\times (j-k)$$
的情况

---

接着，把 $j$ 提出来，发现后面的部分求最大值可以用单调队列，然后就优化了时间复杂度。

代码

#include<algorithm> 
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int f[200010],x[200010];
int q[200010],wz[200010];
int n,m;
struct node
{
	int l,p,s;
}a[200010];
bool cmp(const node&x,const node&y)
{
	return x.s<y.s;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1; i<=m; i++)
	   scanf("%d%d%d",&a[i].l,&a[i].p,&a[i].s);
	sort(a+1,a+1+m,cmp);
	for(int i=1; i<=m; i++)
	 {
	 	int le=max(a[i].s-a[i].l,0),ri=min(a[i].s+a[i].l,n);
	 	int h=1,t=0;
	 	for(int j=le; j<=ri; j++)
	 	   x[j]=f[j]-a[i].p*j;
	 	for(int j=le; j<a[i].s; j++)
	 	 {
	 	 	while(h<=t&&q[t]<=x[j])
	 	 	  t--;
	 	 	q[++t]=x[j];
	 	 	wz[t]=j;
		 }
		for(int j=a[i].s; j<=ri; j++)
		 {
		 	while(h<=t&&wz[h]<j-a[i].l)
	 	 	  h++;
	 	 	if(h>t)
	 	 	  break;
	 	 	f[j]=max(f[j],q[h]+a[i].p*j);
		 }
		for(int j=2; j<=n; j++)
		   f[j]=max(f[j],f[j-1]);
	 }
	printf("%d",f[n]);
	return 0;
}