2021牛客寒假演算法基礎集訓營3 E-買禮物(線段樹區間最大)
阿新 • • 發佈:2021-02-06
題目連結:點這裡~
題目大意
- n個櫃子,每個櫃子禮物編號是,q次詢問。
- 1 x,表示撤走x櫃子裡面的禮物。2 ,表示詢問區間 [xi,yi]是否有兩個相同編號的禮物。
- 對於每次詢問輸出0/1。
思路
- 一看就是資料結構題,但沒想到也是個思維題。
- 對於每個櫃子i,pre[i]表示在i之前禮物編號跟ai一樣的最近櫃子的下標,pos[i]表示在i之後禮物編號跟ai一樣的最近櫃子的下標。
- 對pre[i]構建線段樹,那麼詢問的時候只需要查詢區間最大值,如果這個區間最大值>=,說明區間內有一個禮物前一個位置也在區間內,滿足條件,輸出1,否則輸出0.
- 然後就是區間修改,櫃子x撤回了禮物,那麼線上段樹中的值應該不能發揮作用,那麼我們可以直接給他變成0
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; }