HDU-6047 優先隊列

題目

Maximum Sequence

題意

給兩個數組 {a_i}, {b_i}, 現在要將擴展數組 {a_i} 從 a_n+1 到 a_2n，對於每個新增的 a_i : 必須從 {b_i} 中選出一個 b_k， a_i 需要滿足 a_i ≤ max{a_j - j│b_k ≤ j < i}， b_k 只能被選擇一次。
你需要找出
max{∑2nn+1ai$\sum _{n+1}^{2n}{a}_i$ } modulo 10⁹+7。

題解

優先隊列，簽到題。
根據條件，將每個 a_i 的值更新爲 max {a_j - j | i ≤ j < n}，然後根據{b_i}的值，將他們扔進優先隊列。
最後逐個取出更新 a_n+1 到 a_2n 的值。
需要注意的是，在更新 a_n+2 到 a_2n 值得時候，需要取的值是max{ (a_n+1 - (n+1)) , q.top()}。(a_n+1 - (n+1)) 可能大於優先隊列中的值，但此後的{a_j - j | n+2 ≤ j < n} 一定比 (a_n+1 - (n+1)) 的值小。

代碼

#include <algorithm>
#include <bitset>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <climits>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
using namespace std;
const int N = 250005;
const long long mod = 1000000007;
int a[N];
int b[N];
int main(){
    int n;
    while( cin >> n ){
        for (int i = 1; i <= n; ++i) {
            scanf("%d",&a[i]);
        }
        for (int i = 1; i <= n; ++i) {
            scanf("%d",&b[i]);
        }
        int mx = 0;
        for(int i = n; i >= 1; --i) {
            mx = max(a[i] - i,mx);
            a[i] = mx;
        }
        priority_queue<long long>q;
        for(int i=1;i<=n;i++) {
            q.push((long long)a[b[i]]);
        }
        long long sum = q.top();
        q.pop();
        long long lmx = sum - (n+1);
        while(!q.empty()){
            sum = (sum + max(q.top(),lmx)) % mod;
            q.pop();
        }

        cout << sum << endl;
    }

    return 0;
}