Codeforces Round #219 (Div. 1) C. Watching Fireworks is Fun（单调队列优化dp）

最新推荐文章于 2021-10-30 22:36:27 发布

the best

最新推荐文章于 2021-10-30 22:36:27 发布

阅读量110

点赞数

分类专栏： dp 数据结构

本文链接：https://blog.csdn.net/weixin_45755679/article/details/115152908

版权

dp 同时被 2 个专栏收录

33 篇文章 3 订阅

订阅专栏

数据结构

18 篇文章 1 订阅

订阅专栏

本文介绍了如何运用动态规划和单调队列优化来解决一个关于烟花燃放时的最大收益问题。给定一定数量的点和烟花，每个烟花在特定时间和地点燃放，目标是在烟花燃放时最大化收益，而每秒最多能移动一定距离。通过建立状态转移方程并利用单调队列维护最大值，实现了高效求解。

摘要由CSDN通过智能技术生成

题目链接
题意：

给出一条线段，有 $n$ 个点，编号为 $1$ 到 $n$ ，有 $m$ 个烟花，第 $i$ 个烟花会在 $t_i$ 时刻在 $a_i$ 处燃放，假如此时你的位置为 $x$ ，那么收益就是 $b_i-|a_i-x|$ 。每秒最多可以移动 $d$ 单位，求为能获得的最大收益。

题解：

先考虑 $d p$ ,设 $d p [i] [j]$ 表示第 $i$ 个烟花燃放时处在 $j$ 位置的最大收益,那么可以写出转移方程：

$dp[i][j]= \max (dp[i-1][k]) +b_i-|a_i-j|$

$\leq k \leq j+(t[i]-t[i-1])*d$

很明显，这肯定不能暴力求解 $d p [i - 1] [k]$ ,既然是要找最大值，那么可以利用单调队列维护最大值即可。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=2e5+5;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int n,m,d;
struct node
{
    ll a,b,t;
}f[MAXN];
ll dp[2][MAXN];
ll q[MAXN];
bool cmp(node x,node y)
{
    return x.t<y.t;
}
int main()
{
    memset(dp,-inf,sizeof dp);
    cin>>n>>m>>d;
    for(int i=1;i<=m;i++)
    {
        cin>>f[i].a>>f[i].b>>f[i].t;
    }
    sort(f+1,f+1+m,cmp);
    int fl=1;
    for(int i=1;i<=n;i++) dp[0][i]=0;
    for(int i=1;i<=m;i++)
    {
        int k=1;
        int l=1,r=0;
        for(int j=1;j<=n;j++)
        {
            ll maxx=j+(f[i].t-f[i-1].t)*d;
            if(maxx>n) maxx=n;
            ll minn=j-(f[i].t-f[i-1].t)*d;
            if(minn<1) minn=1;
            while(k<=maxx)
            {
                while(l<=r&&dp[fl^1][k]>dp[fl^1][q[r]]) r--;
                q[++r]=k;
                k++;
            }
            while(l<=r&&q[l]<minn) l++;
            dp[fl][j]=dp[fl^1][q[l]]+f[i].b-abs(f[i].a-j);
        }
        fl=(fl^1);
    }
    ll ans=-1e18;
    for(int i=1;i<=n;i++)
    {
        ans=max(ans,dp[fl^1][i]);
    }
    printf("%lld\n",ans);
}