1. 程式人生 > 其它 >2021牛客寒假演算法基礎集訓營3 E-買禮物(線段樹區間最大)

2021牛客寒假演算法基礎集訓營3 E-買禮物(線段樹區間最大)

題目連結:點這裡~

題目大意

  • n個櫃子,每個櫃子禮物編號是a_i,q次詢問。
  • 1 x,表示撤走x櫃子裡面的禮物。2l_i r_i,表示詢問區間 [xi,yi]是否有兩個相同編號的禮物。
  • 對於每次詢問輸出0/1。

思路

  • 一看就是資料結構題,但沒想到也是個思維題。
  • 對於每個櫃子i,pre[i]表示在i之前禮物編號跟ai一樣的最近櫃子的下標,pos[i]表示在i之後禮物編號跟ai一樣的最近櫃子的下標。
  • 對pre[i]構建線段樹,那麼詢問的時候只需要查詢區間最大值,如果這個區間最大值>=l_i,說明區間內有一個禮物前一個位置也在區間內,滿足條件,輸出1,否則輸出0.
  • 然後就是區間修改,櫃子x撤回了禮物,那麼線上段樹中的值應該不能發揮作用,那麼我們可以直接給他變成0
    ,影響不了別人。於此同時,我們還要更新pre[x]這個櫃子j,使得pos[j]不再是x,而是pos[x],這相當於一個連結串列刪除的操作。

ac程式碼

#include<bits/stdc++.h>
using namespace std;
#define io cin.tie(0);ios::sync_with_stdio(false);
#define debug(x) cout<<#x<<"="<<x<<endl
#define lowbit(x) x&(-x)
#define pii pair<int,int>
#define mk make_pair
#define ll long long
#define rs p<<1|1
#define ls p<<1
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
inline ll read(){
    ll p=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){p=(p<<1)+(p<<3)+(c^48),c=getchar();}
    return f*p;
}

struct node{
    int l, r;
    int lazy, sum, mx;
}tree[maxn << 2];
int a[maxn], pre[maxn], pos[maxn];
void pushup(int p){
    tree[p].mx = max(tree[ls].mx, tree[rs].mx);
}

void build(int p, int l, int r){
    tree[p].l = l; tree[p].r = r;
    if(l == r){
        tree[p].mx = pre[l];
        return;
    }
    int mid = l + r >> 1;
    build(ls, l, mid);
    build(rs, mid + 1, r);
    pushup(p);
}
void update(int p, int l, int r, int val) {//區間的每個值都改為val
    if(l <= tree[p].l && tree[p].r <= r){
        tree[p].mx = val;
        return;
    }
    int mid = tree[p].l + tree[p].r >> 1;
    if(mid >= l) update(ls, l, r, val);
    if(mid + 1 <= r) update(rs, l, r, val);
    pushup(p);
}
int query(int p, int l, int r){
    if(l <= tree[p].l && tree[p].r <= r)
        return tree[p].mx;
    int ans = -inf;
    int mid = tree[p].l + tree[p].r >> 1;
    if(mid >= l) ans = max(ans, query(ls, l, r));
    if(mid + 1 <= r) ans = max(ans, query(rs, l, r));
    return ans;
}
void solve(){
    int n, m;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++){
        a[i] = pre[i] = 0;
        pos[i]= n + 1;
    }
    for(int i = 1; i <= n; i ++){
        int x; scanf("%d", &x);
        pos[a[x]] = i;
        pre[i] = a[x];
        a[x] = i; //當前x出現的最近位置
    }
    build(1, 1, n);
    while(m --){
        int op, x;
        scanf("%d", &op);
        if(op == 1){
            scanf("%d", &x);
            update(1, x, x, 0);
            if(pos[x] <= n){
                update(1, pos[x], pos[x], pre[x]); //如果禮物ax後面還有的話更新
                pos[pre[x]] = pos[x];
            }
        }else{
            int l, r;
            scanf("%d%d", &l, &r);
            printf("%d\n", query(1, l, r) >= l);
        }
    }
}
int main(){
    solve();
    return 0;
}