[数据结构与算法]POJ - 1821 Fence（单调队列优化DP）

POJ - 1821 Fence（单调队列优化DP）

#include<cstdio>
#include<algorithm>
#include<deque>
using namespace std;
const int N=110,M=16010;
struct Node
{
    int p,l,s;//单价;最高长度;必备点
    bool operator <(const Node& t) const {return s<t.s;}
}a[M];
int f[N][M];
deque<int> q;
int date(int i,int k)
{
    return f[i-1][k]-a[i].p*k;
}
int main()
{
    int n,m;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);
	for(int i=1;i<=m;i++)//粉刷匠
    {
        for(int k=max(0,a[i].s-a[i].l);k<a[i].s;k++)
        {
            while (q.size() && date(i,q.back())<=date(i,k)) q.pop_back();//不是最优值
            q.push_back(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(q.size() && q.front()<j-a[i].l)//不在候选范围内了.
                    q.pop_front();
                if(q.size()) f[i][j]=max(f[i][j],a[i].p*j+date(i,q.front()));//计算
            }
        }
    }
    printf("%d\n",f[m][n]);
    return 0;
}