A Magic Lamp HDU - 3183

这个题方法很简单啊,可是没有考虑
0 1
这个例子,就wr了整整一下午
写个博客纪念一下

/*
线段树维护区间最小值
贪心:每次找可行区域的最小值
*/
#include
#include
#include
#include
using namespace std;
const int inf=0x3f3f3f3f;
const int num=8*1000+10;
struct node
{
    int L,R,mi,mp;//mi最小值,mp最小值位置
}a[num];
int n,m;
char s[num],d[num];
int p,mi;
void update(int k)
{
    int t;
    t= (a[k<<1].mi <= a[k<<1|1].mi) ? (k<<1) : (k<<1|1);
    a[k].mi=a[t].mi;
    a[k].mp=a[t].mp;
}
void build(int k,int L,int R)
{
    a[k].L=L;a[k].R=R;a[k].mi=0;a[k].mp=L;
    if(L==R){
        a[k].mi=s[L-1]-'0';
        return ;
    }
    int mid=(L+R)>>1,t=k<<1;
    build(t,L,mid);
    build(t|1,mid+1,R);
    update(k);
    return ;
}
void getnum(int k,int L,int R)
{
    if(a[k].L>=L&&a[k].R<=R){
        if(a[k].mi<mi) {
            mi=a[k].mi;
            p=a[k].mp;
        }
        else if(a[k].mi==mi&&a[k].mp<p)
        {
        	p=a[k].mp;
		}
		
        return ;
    }
    int mid=(a[k].L+a[k].R)>>1;
    if(R>mid)  getnum(k<<1|1,L,R);
    if(L<=mid) getnum(k<<1,L,R);
    return ;
}
int main()
{
    while(scanf("%s %d",s,&m)!=EOF)
    {
        n=strlen(s);
        build(1,1,n);
        int k=n-m,cnt=0;
        
        p=0;
        
        while(k>0)
        {
        	mi=inf;
            getnum(1,p+1,n-k+1);
            d[++cnt]=s[p-1];
            k--;
        }
        
        d[cnt+1]='\0';
        
        int c=1;
        for(int i=1;i<cnt;i++)
        {
            if(d[i]=='0') c++;
            else break;
        }
        if(c>cnt){
        	printf("0\n");
		}
		else 
        printf("%s\n",d+c);
    }
    return 0;
}

你可能感兴趣的:(线段树,rmq)