1. 程式人生 > >【NOIP2016提高A組五校聯考4】ksum

【NOIP2016提高A組五校聯考4】ksum

分析 -s down wap algo include CA namespace LG

題目

技術分享圖片

分析

發現,當子段[l,r]被取了出來,那麽[l-1,r]、[l,r+1]一定也被取了出來。
那麽,首先將[1,n]放入大頂堆,每次將堆頂的子段[l,r]取出來,因為它是堆頂,所以一定是最大的子段,輸出它,並將[l+1,r]和[l,r-1]放進堆中。
一共就只用做k次就可以了。

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
const int maxlongint=2147483647;
const int mo=1000000007;
const int N=100005;
using namespace std;
long long n,m,a[N],sum[N],t[N*3][3],tot=1;
int swap(int x,int y,int z)
{
    t[x][z]=t[x][z]^t[y][z];
    t[y][z]=t[x][z]^t[y][z];
    t[x][z]=t[x][z]^t[y][z];
}
int up(int x)
{
    while(t[x][2]>t[x/2][2] && x!=1)
    {
        for(int i=0;i<=2;i++)
            swap(x,x/2,i);
        x/=2;
    }
}
int down(int x)
{
    int s=x*2,s1=x*2+1;
    while((t[x][2]<t[s][2] || t[x][2]<t[s1][2]) && s<=tot)
    {
        if(t[x][2]>t[s][2] && s1>tot) break;
        if(t[s][2]>t[s1][2])
        {
            for(int i=0;i<=2;i++)
                swap(x,s,i);
            x=s;
        }
        else
        if(s1<=tot)
        {
            for(int i=0;i<=2;i++)
                swap(x,s1,i);
            x=s1;
        }
        s=x*2,s1=x*2+1;
    }
}
int insert(int x,int y)
{
    t[++tot][0]=y;
    t[tot][1]=x;
    t[tot][2]=sum[x]-sum[y-1];
    up(tot);
}
int cut()
{
    t[1][0]=t[tot][0];
    t[1][1]=t[tot][1];
    t[1][2]=t[tot--][2];
    t[tot+1][0]=t[tot+1][1]=t[tot+1][2]=0;
    down(1);
}
int main()
{
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    tot=1;
    t[tot][0]=1;
    t[tot][1]=n;
    t[tot][2]=sum[n]-sum[0];
    while(m--)
    {
        printf("%lld ",t[1][2]);
        int x=t[1][1],y=t[1][0]+1,x1=t[1][1]-1,y1=t[1][0];
        cut();
        if(x==n) if(x>=y) if(y<=n) insert(x,y);
        if(x1>=y1) if(x1>=1) insert(x1,y1);
    }
}

【NOIP2016提高A組五校聯考4】ksum