HaiHongOJ 1003 God Wang

题目连接：http://oj.haihongblog.com/problem.php?id=1003

线段树维护区间最小值，并且求解下标

#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;

const int maxn=+ ;
const int INF=0x7FFFFFFF;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int N,K,zzt;
vector<int> ans;

int MIN[maxn<<];  

void PushUP(int rt)
{
    MIN[rt]=min(MIN[rt<<],MIN[rt<<|]);
}

void Build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%d",&MIN[rt]);
        return;
    }
    int m=(l+r)>>;
    Build(lson);
    Build(rson);
    PushUP(rt);
}

void Update(int p,int add,int l,int r,int rt)
{
    if(l==r)
    {
        MIN[rt]=add;
        return;
    }
    int m=(l+r)>>;
    if(p<=m)
        Update(p,add,lson);
    else
        Update(p,add,rson);
    PushUP(rt);
}

int Query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r) return MIN[rt];
    int m=(l+r)>>;
    int ret=INF;
    if(L<=m)   ret=min(ret,Query(L,R,lson));
    if(R>m)    ret=min(ret,Query(L,R,rson));
    return ret;
}

void Find(int L,int R,int l,int r,int rt)
{

    if(L<=l&&R>=r)
    {
        int lx=l,rx=r,node=rt;
        while()
        {
            int mx=(lx+rx)/;
            if(lx==rx) {zzt=lx;break;}
            if(K>=MIN[*node])
            {
                node=*node;
                rx=mx;
                continue;
            }

            if(K>=MIN[*node+])
            {
                node=*node+;
                lx=mx+;
                continue;
            }
        }
        return;
    }

    int m=(l+r)>>;
    if(K>=MIN[rt<<]&&L<=m)     Find(L,R,lson);
    if(zzt!=-) return;
    if(K>=MIN[rt<<|]&&R>m) Find(L,R,rson);
    if(zzt!=-) return;
}

int main()
{
    //    freopen("F:\\test (1).in","r",stdin);
    //    freopen("F:\\out.txt","w",stdout);

    while(~scanf("%d%d",&N,&K)){
        Build(,N,);

        bool fail=;
        int pos=;
        ans.clear();

        for(int i=;i<=N;i++)
        {
            int P=Query(pos,N,,N,);
            if(K>=P) //后面一段存在可以吃的点
            {
                zzt=-;

                //找到第一个大于等于K的位置 zzt
                Find(pos,N,,N,);
                ans.push_back(zzt);
                pos=zzt;
                //该位置的值更新为INF
                Update(zzt,INF,,N,);
                K++;
            }
            else //后面一段不存在可以吃的点，从头开始找
            {
                P=Query(,N,,N,);
                if(K>=P)
                {
                    zzt=-;

                    //找到第一个大于等于K的位置 zzt
                    Find(,N,,N,);
                    ans.push_back(zzt);
                    pos=zzt;
                    //该位置的值更新为INF
                    Update(zzt,INF,,N,);
                    K++;
                }

                else //找不到了，失败
                {
                    fail=;
                    break;
                }
            }
        }

        if(fail) printf("-1\n");
        else
        {
            for(int i=;i<ans.size();i++)
            {
                printf("%d",ans[i]);
                if(i<ans.size()-) printf(" ");
                else printf("\n");
            }
        }
    }
    return ;
}