2018年ACM俱乐部个人训练赛第十五场 问题 C: 飞扬的小鸟II（dp+优先队列优化）

问题 C: 飞扬的小鸟II

时间限制: 1 Sec   内存限制: 128 MB

题目描述

有n棵大树从左到右排成一排，编号为1到n，每棵有高度hi与疲劳值wi。 有一只鸟儿现在站在最左侧的1号大树上，它想飞到第n棵树上去，但它不 能连续飞行太远，当它在第i棵树上时，只能飞到第i + 1, i + 2,..., i + k棵 树上，并获得对应大树的疲劳值；同时，假如它飞到的那棵树的高度不小 于当前这棵树的高度，那么它会获得额外的d疲劳值。求飞到第n棵树的最 小疲劳值是多少。

输入

第一行三个整数n、k和d，由空格分割，表示树有多少棵，飞行的最大距离，以及额外的疲劳值。 接下来n行，第i行包含两个整数hi和wi，表示第i棵树的高度和疲劳值。

输出

输出一行，表示最小疲劳值。

样例输入

5 2 1

4 3

3 2

5 3

2 1

6 1

样例输出

8

提示

最优策略是1 → 2 → 4 → 5，总疲劳值是3 + 2 + 1 + 1，并在4 → 5时支付额外的1疲劳值。

对于20%的数据，n ≤ 1000；
对于另外40%的数据，d = 0；
对于100%的数据，1 ≤ k < n ≤ 3 ∗ 105, 0 ≤ hi ≤ 109, 0 ≤ wi ≤ 103, d ∈{0, 1}。

分析：

直接dp复杂度O(n*k)  　状态转移方程 dp[i]=min(dp[i],dp[k]+a[i].wi+(a[k].hi>a[i].hi?0:d)) (i-k=<k<=i-1)

这样会超时　　　第二维可以优化一下　　用优先队列维护一下这个区间的最小值就行

AC代码：

#include <bits/stdc++.h>
#define mset(a,x) memset(a,x,sizeof(a))
typedef long long ll;
using namespace std;
struct node{
    ll hi,wi;
    int Index;
    friend bool operator < (node a, node b){
        if(a.wi==b.wi) return a.hi<b.hi;
        return a.wi>b.wi;
    }
}a[3000005];
int main (){
    int n,k,d;
    while (scanf ("%d%d%d",&n,&k,&d)!=EOF){
        priority_queue<node> Q;
        for (int i=1;i<=n;i++) scanf ("%lld%lld",&a[i].hi,&a[i].wi),a[i].Index=i;
        Q.push(a[1]);
        for (int i=2;i<=n;i++){
            while(!Q.empty()){
                if(i-Q.top().Index>k) Q.pop();
                else{
                    a[i].wi=a[Q.top().Index].wi+a[i].wi+(Q.top().hi>a[i].hi?0:d);
                    break;
                }
            }
            Q.push(a[i]);
        }
        printf("%lld\n",a[n].wi);
    }
    return 0;
}