1150: [CTSC2007]数据备份Backup

1150: [CTSC2007]数据备份Backup

https://lydsy.com/JudgeOnline/problem.php?id=1150

分析：

　　堆+贪心。

　　每次选最小的并一定是最优的，如果不选这个最小的，那一定是为了取它左右两边（两条都要取才可能比当前优）。

　　如果先选了最小的，考虑后面如何撤销。选了这个后，左右两边的线段就要删了，那么在加入一个长度为：左边的长度+右边的长度-当前的线段的长度 的一条线段。

　　wqs二分，模拟费用流

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<cctype>
#include<set>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long LL;

inline int read() {
    int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for(;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f;
}

const int N = 100005;
const int INF = 1e9;

int pre[N], nxt[N], d[N], a[N];
bool vis[N]; 

struct Edge{
    int len, id;
    Edge() { }
    Edge(int a, int b) { len = a, id = b; }
    bool operator < (const Edge &A) const {
        return len > A.len;
    }
};
priority_queue<Edge> q;

void del(int i) {
    vis[i] = 1;
    nxt[pre[i]] = nxt[i];
    pre[nxt[i]] = pre[i];
}

int main() {
    int n = read(), m = read();
    for (int i = 1; i <= n; ++i) d[i] = read();
    for (int i = 1; i < n; ++i) {
        a[i] = d[i + 1] - d[i];
        q.push(Edge(a[i], i));
    }
    a[0] = a[n] = INF;
    for (int i = 1; i <= n; ++i) pre[i] = i - 1, nxt[i] = i + 1;
    int ans = 0;
    while (m --) {
        while (vis[q.top().id]) q.pop();
        Edge now = q.top(); q.pop();
        ans += now.len;
        int i = now.id, t;
        t = min(a[pre[i]] + a[nxt[i]] - now.len, INF);
        del(pre[i]); del(nxt[i]);
        a[i] = t; q.push(Edge(a[i], i));
    }
    cout << ans;
    return 0;
}