【ybtoj 高效进阶 5.6】 【单调队列】 粉刷木板
题目
解题思路
设f[i][j]为前i个工匠刷到第j块板
那么第i个工匠可以什么都不干 f[i][j]=f[i-1][j]
或者不刷第j块 f[i][j]=f[i][j-1]
要不刷了一段[k+1,j]的区间 f[i][j]=max(f[i][j],f[i][k]+p[i]*(j-k))
很开心得发现TLE 啧
把式子化开得 f[i][j]=max(f[i][j],f[i][k]+p[i]*j-p[i]*k)
得出p[i]*j是固定的,求f[i][k]-p[i]*k的最大值即可
拿单调队列维护
代码
#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
struct lzf{
int l,p,s;
}a[120];
int n,m,f[120][16010],q[16010];
bool cmp(lzf l,lzf y)
{
return l.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+m+1,cmp);
for (int i=1;i<=m;i++)
{
int t=0,h=1;
for (int k=max(a[i].s-a[i].l,0);k<a[i].s;k++)
{
while (h<=t&&f[i-1][q[t]]-a[i].p*q[t]<=f[i-1][k]-a[i].p*k) t--;
q[++t]=k;
}
for (int j=1;j<=n;j++)
{
f[i][j]=max(f[i-1][j],f[i][j-1]);
if (j>=a[i].s)
{
while (h<=t&&j-q[h]>a[i].l) h++;
if (h<=t) f[i][j]=max(f[i][j],f[i-1][q[h]]+(j-q[h])*a[i].p);
}
}
}
printf("%d",f[m][n]);
return 0;
}