1. 程式人生 > >BZOJ 2006 淺談資料結構優化貪心思路

BZOJ 2006 淺談資料結構優化貪心思路

這裡寫圖片描述
世界真的很大
今天考試考了NOI2010年的題
好不容易想出來了第二題,就是這道,在加上第一題會做,一共200分
認真分析題意比較重要,就算看出來是貪心也不見得能過,想辦法從貪心的性質入手,去優化

看題先:

description

小Z是一個小有名氣的鋼琴家,最近C博士送給了小Z一架超級鋼琴,小Z希望能夠用這架鋼琴創作出世界上最美妙的
音樂。 這架超級鋼琴可以彈奏出n個音符,編號為1至n。第i個音符的美妙度為Ai,其中Ai可正可負。 一個“超級
和絃”由若干個編號連續的音符組成,包含的音符個數不少於L且不多於R。我們定義超級和絃的美妙度為其包含的
所有音符的美妙度之和。兩個超級和絃被認為是相同的,當且僅當這兩個超級和絃所包含的音符集合是相同的。 
小Z決定創作一首由k個超級和絃組成的樂曲,為了使得樂曲更加動聽,小Z要求該樂曲由k個不同的超級和絃組成。
我們定義一首樂曲的美妙度為其所包含的所有超級和絃的美妙度之和。小Z想知道他能夠創作出來的樂曲美妙度最
大值是多少。

input

第一行 包含四個正整數n, k, L, R。其中n為音符的個數,k為樂曲所包含的超級和絃個數,LR分別是超級和絃所
包含音符個數的下限和上限。 接下來n行,每行包含一個整數Ai,表示按編號從小到大每個音符的美妙度。
N<=500,000
k<=500,000
-1000<=Ai<=1000,1<=L<=R<=N且保證一定存在滿足條件的樂曲

output

只有一個整數,表示樂曲美妙度的最大值。

首先分析題意
一段和絃指的是原序列上連續的一段
考慮樂曲由和絃組成,不會出現和絃相交產生新和絃的情況
等價於和絃不同只有每個和絃都不同。
那麼題意就等價於把原圖的所有“段”排個序,選取前K個
但是這樣肯定會T

考慮選擇不同“段”,當且僅當左端點或者右端點不同,所以只需要確定“段”不同,只需要確定左端點不同就可以唯一定“段”的不同了

而只要左端點確定了,右端點雖然不能確定,但是右端點的範圍就可以確定了,我們就把每個左端點確定的區間右端點範圍內最大的區間選出來

但是隻是這樣的話就有些問題,有可能在同一個區間裡存在多個在前K個裡面的“段”,這樣的話剩下的點就會選不到

而考慮對於每一個固定的左端點,只有右端點範圍內最大的那個值被選了,次大的才會被選,所以完全沒有必要一開始就加進去
所以就記錄每一個最大值的為值,選了他之後把他的左邊和右邊的區間加進去

又考慮每次排一次序的時間複雜度實在太大,而我們其實並不關心每次的“段”的順序,只需要最大值罷了,所以把排序改成堆

重複K次上述操作,累加就是答案

完整程式碼:

#include<stdio.h>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long dnt;

struct music
{
    int st,lf,rg,tar;
    dnt w;
};

int a[1000010],n,L,R,K;
dnt sum[1000010],ans=0;

bool operator <(const music &r, const music &s) 
{
    return r.w < s.w;
}

priority_queue < music > state;

struct node
{
    int pos;
    node *ls,*rs;
    void update()
    {
        if(sum[ls->pos]>=sum[rs->pos]) pos=ls->pos;
        else pos=rs->pos;
    }
}pool[8000010],*tail=pool,*root;

node *build(int lf,int rg)
{
    node *nd=++tail;
    if(lf==rg)
    {
        nd->pos=lf;
        nd->ls=nd->rs=0;
        return nd;
    }
    int mid=(lf+rg)>>1;
    nd->ls=build(lf,mid);
    nd->rs=build(mid+1,rg);
    nd->update();
    return nd;
}

int query(node *nd,int lf,int rg,int L,int R)
{
    if(L<=lf&&rg<=R)
        return nd->pos;
    int mid=(lf+rg)>>1,rt=0,lt=0;
    nd->update();
    if(L<=mid) lt=query(nd->ls,lf,mid,L,R);
    if(R>mid) rt=query(nd->rs,mid+1,rg,L,R);
    if(!rt) return lt;
    if(!lt) return rt;
    if(sum[rt]>sum[lt]) return rt;
    return lt;
}

int main()
{
    scanf("%d%d%d%d",&n,&K,&L,&R);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i];
    root=build(1,n);
    for(int i=1;i<=n-L+1;i++)
    {
        music tmp;
        int x=query(root,1,n,i+L-1,min(i+R-1,n));
        tmp.st=i,tmp.lf=i+L-1,tmp.rg=min(i+R-1,n);
        tmp.w=sum[x]-sum[i-1],tmp.tar=x;
        state.push(tmp);
    }
    while(K--)
    {
        music tmp=state.top();
        state.pop();
        ans+=tmp.w;
        music aa,bb;
        aa.st=tmp.st,aa.lf=tmp.lf,aa.rg=tmp.tar-1;
        bb.st=tmp.st,bb.lf=tmp.tar+1,bb.rg=tmp.rg;
        int x,y;
        if(aa.lf<=aa.rg  )
        {
            x=query(root,1,n,aa.lf,min(n,aa.rg));
            aa.w=sum[x]-sum[aa.st-1];
            aa.tar=x,state.push(aa);
        }
        if(bb.lf<=bb.rg  )
        {
            y=query(root,1,n,bb.lf,min(n,bb.rg));
            bb.w=sum[y]-sum[bb.st-1];
            bb.tar=y,state.push(bb);
        }
    }
    cout << ans << endl;
    return 0;
}

嗯,就是這樣