BZOJ1150 [CTSC2007] 数据备份Backup 贪心_堆_神题

最新推荐文章于 2019-05-22 10:28:00 发布

原创最新推荐文章于 2019-05-22 10:28:00 发布 · 123 阅读

0 ·

CC 4.0 BY-SA版权

博客围绕BZOJ1150 [CTSC2007] 数据备份Backup题目展开，指出这是一道非常神的题，涉及贪心和堆的知识。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

BZOJ1150 [CTSC2007] 数据备份Backup 贪心_堆_神题

非常神的一道题.

Code:

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 1000000 
#define inf 100000000000
using namespace std; 
struct Node{
    long long  key;
    int id; 
    Node(long long a=0,int b=0):key(a),id(b){} 
    bool operator < ( Node c) const{
        return key > c.key; 
    }
}; 
priority_queue<Node>Q;  
int tag[maxn],suf[maxn],pre[maxn];       
long long h[maxn],arr[maxn];  
int main(){
    // setIO("input");   
    int n,k;
    scanf("%d%d",&n,&k); 
    for(int i=1;i<=n;++i) scanf("%lld",&arr[i]); 
    for(int i=1;i<=n;++i)  { h[i]=(long long) arr[i]-arr[i-1], suf[i]=i+1,pre[i]=i-1; } 
    h[0]=inf, suf[n]=pre[2]=0; 
    for(int i=2;i<=n;++i) Q.push(Node(h[i],i));        
    long long ans = 0; 
    for(int i=1;i<=k;++i) { 
        while(tag[Q.top().id]) Q.pop();  
        Node u = Q.top(); 
        ans += (long long) u.key;        
        int cur = u.id;  
        int l = pre[cur],r = suf[cur];      
        h[cur] = h[l]+h[r]-h[cur];                  
        tag[l]=tag[r]=1; 
        h[l]=h[r]=inf; 
        pre[cur]=pre[l],suf[pre[cur]] = cur; 
        suf[cur]=suf[r],pre[suf[r]] = cur;     
        Q.pop(); 
        Q.push(Node(h[cur],cur));                      
    } 
    printf("%lld",ans); 
    return 0; 
}