[題解]AT_abc106_d [ABC106D] AtCoder Express 2

WaterSunHB發表於2024-06-22

思路

問題本質上就是一個在一段區間中找完整線段的數量。

我們先不考慮所有 \(l_i\) 對答案的限制,那麼,我們的答案就應該是 \(1 \sim q\) 線段的數量減去 \(1 \sim (p - 1)\) 的數量,這個東西可以直接用樹狀陣列維護。

然後,再來考慮 \(l_i\) 對答案的限制。如果線段 \(i\) 能對答案產生貢獻,當且僅當 \(p \leq l_i \wedge r_i \leq q\)

因此,我們如果每次只將 \(l_i \geq p\) 的線段加入樹狀陣列,就可以保證答案正確了,但是時間複雜度是 \(\Theta(qm \log n)\),考慮最佳化。

如果我們對於每一次查詢時對於樹狀陣列的操作只加不減,那麼,時間複雜度可以最佳化成 \(\Theta(q \log n)\)

於是我們直接離線搞,將線段和詢問都按照左端點為第一關鍵字從大到小排序,那麼,我們在前一次詢問存在在樹狀陣列的元素就不用刪除,只需要將新的元素加入即可。

Code

#include <bits/stdc++.h>  
#define re register  
  
using namespace std;  
  
const int N = 510,M = 2e5 + 10;  
int n,m,q;  
int ans[M];  
  
struct array{  
    int l;  
    int r;  
  
    bool operator <(const array &t)const{  
        return l > t.l;  
    }  
}arr[M];  
  
struct point{  
    int l;  
    int r;  
    int id;  
  
    bool operator <(const point &t)const{  
        return l > t.l;  
    }  
}Q[M];  
  
struct BIT{  
    int tr[N];  
  
    inline int lowbit(int x){  
        return x & -x;  
    }  
  
    inline void modify(int x,int k){  
        for (re int i = x;i <= n;i += lowbit(i)) tr[i] += k;  
    }  
  
    inline int query(int x){  
        int res = 0;  
        for (re int i = x;i;i -= lowbit(i)) res += tr[i];  
        return res;  
    }  
}tree;  
  
inline int read(){  
    int r = 0,w = 1;  
    char c = getchar();  
    while (c < '0' || c > '9'){  
        if (c == '-') w = -1;  
        c = getchar();  
    }  
    while (c >= '0' && c <= '9'){  
        r = (r << 1) + (r << 3) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
int main(){  
    n = read();  
    m = read();  
    q = read();  
    for (re int i = 1;i <= m;i++){  
        arr[i].l = read();  
        arr[i].r = read();  
    }  
    for (re int i = 1;i <= q;i++){  
        Q[i].l = read();  
        Q[i].r = read();  
        Q[i].id = i;  
    }  
    sort(arr + 1,arr + m + 1);  
    sort(Q + 1,Q + q + 1);  
    int idx = 1;  
    for (re int i = 1;i <= q;i++){  
        int ql = Q[i].l;  
        int qr = Q[i].r;  
        while (arr[idx].l >= ql && idx <= m){  
            tree.modify(arr[idx].r,1);  
            idx++;  
        }  
        ans[Q[i].id] = tree.query(qr) - tree.query(ql - 1);  
    }  
    for (re int i = 1;i <= q;i++) printf("%d\n",ans[i]);  
    return 0;  
}  

相關文章