題目鏈接

題意：數字字符串刪除k個值后的數字最小不算前導0

思路：其實就是最小字典序，那么我們肯定一位一位的確定最小的值，那么對于i位，其實找i+1~i+k位中最小的數有沒有比當前位置的數小，如果有的話就是把中間的都刪掉。所以其實就是求某一位i右邊i+1~i+1+k中最小值的最小下標，我們可以用線段樹來維護。哈哈哈~問了問學長，學長罵我智障(hh

其實就是線段樹維護個node，重載一下就行~

我看題解居然還有n方的過的，太可惜了

AC代碼：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#include<time.h>#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a, b, c, d) min(min(a,b),min(c,d))
#define min3(x, y, z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
#define li __int128
//void print(li x) {if(x>9) print(x/10);putchar(x%10+48);}
const int dir[4][2] = {0, -1, -1, 0, 0, 1, 1, 0};
typedef long long ll;
typedef unsigned long long ull;
const ll inFF = 9223372036854775807;
const int maxn=1e5+5;
const ll mod=1e9+7;
using namespace std;
int a[maxn];
char s[maxn];
struct node
{int val,pos;bool friend operator<(node s,node e){if(s.val==e.val)  return s.pos<e.pos;return s.val<e.val;}
}tree[maxn<<1];
int n,k,vis[maxn];
void pushup(int i)
{tree[i]=(tree[lson]<tree[rson])?tree[lson]:tree[rson];
}
void build(int i,int l,int r)
{if(l==r){tree[i]=node{a[l],l};return;}int mid=half;build(Lson);build(Rson);pushup(i);
}
node query(int i,int l,int r,int ql,int qr)
{if(ql<=l&&qr>=r) return tree[i];int mid=half;if(qr<=mid) return query(Lson,ql,qr);else if(ql>mid) return query(Rson,ql,qr);else{node s=query(Lson,ql,mid);node e=query(Rson,mid+1,qr);if(s<e) return s;else return e;}
}
int main()
{while(scanf("%s",s+1)!=EOF){n=strlen(s+1);for(int i=1;i<=n;i++) a[i]=s[i]-'0',vis[i]=0;cin>>k;build(1,1,n);int c=k;for(int i=1;i<=n&&c>0;i++){if(c>=n-i+1){for(int j=i;j<=n;j++)vis[j]=1;break;}if(a[i]==0){vis[i]=0;continue;}node now=query(1,1,n,i+1,min(i+c,n));if(now.val<a[i]){vis[now.pos]=0;for(int j=i;j<now.pos;j++) vis[j]=1;c-=now.pos-i;i=now.pos;}else vis[i]=0;}int num=0,fg=0;for(int i=1;i<=n;i++){if(!vis[i]){if(a[i]==0&&fg==0) continue;printf("%d",a[i]),num++,fg=1;}}if(num==0) puts("0");else printf("\n");}return 0;
}

?

總結

以上是生活随笔為你收集整理的HDU - 3183 A Magic Lamp 线段树的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。