BGSHOOT - Shoot and kill 离散化 树状数组或前缀和 ST表 模板题

先对出现的时间点离散化。

树状数组维护前缀和，表示n种动物的出没。

ST查询区间最值。

wa了很多次，其实离散化的时候把x、y、l、r全都加进去，会省很多事。

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
typedef unsigned int uii;
const int maxn=100005;
int n,q,l[maxn],r[maxn],st[19][maxn*4],x[maxn],y[maxn],tx,ty,bit[maxn*4];
vector<int> v;
void add(int x,int t) {
    while ((uii)x<=v.size()) {
        bit[x]+=t;
        x+=x&-x;
    }
}
int sum(int x) {
    int ret=0;
    while (x) {
        ret+=bit[x];
        x-=x&-x;
    }
    return ret;
}
void init() {
    for (uii i=1;i<=v.size();++i)
        st[0][i]=sum(i);
    for (int i=1;i<19;++i)
        for (uii j=1;j+(1<<i)-1<=v.size();++j)
            st[i][j]=max(st[i-1][j],st[i-1][j+(1<<(i-1))]);
}
int query(int l,int r) {
    int i=0;
    while ((1<<i)<=r-l+1)
        ++i;
    --i;
    return max(st[i][l],st[i][r-(1<<i)+1]);
}
int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;++i) {
        scanf("%d%d",&l[i],&r[i]);
        v.push_back(l[i]);
        v.push_back(r[i]);
    }
    scanf("%d",&q);
    for (int i=0;i<q;++i) {
        scanf("%d%d",&x[i],&y[i]);
        v.push_back(x[i]);
        v.push_back(y[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for (int i=0;i<n;++i) {
        add(lower_bound(v.begin(),v.end(),l[i])-v.begin()+1,1);
        add(lower_bound(v.begin(),v.end(),r[i])-v.begin()+2,-1);
    }
    init();
    for (int i=0;i<q;++i) {
        tx=lower_bound(v.begin(),v.end(),x[i])-v.begin()+1;
        ty=lower_bound(v.begin(),v.end(),y[i])-v.begin()+1;
        printf("%d\n",query(tx,ty));
    }
    return 0;
}

由于可以离线处理，树状数组可以拿掉，维护一个前缀和就行了。

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
typedef unsigned int uii;
const int maxn=100005;
int n,q,l[maxn],r[maxn],st[19][maxn*4],x[maxn],y[maxn],tx,ty;
vector<int> v;
void init() {
    for (uii i=1;i<=v.size();++i)
        st[0][i]+=st[0][i-1];
    for (int i=1;i<19;++i)
        for (uii j=1;j+(1<<i)-1<=v.size();++j)
            st[i][j]=max(st[i-1][j],st[i-1][j+(1<<(i-1))]);
}
int query(int l,int r) {
    int i=0;
    while ((1<<i)<=r-l+1)
        ++i;
    --i;
    return max(st[i][l],st[i][r-(1<<i)+1]);
}
int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;++i) {
        scanf("%d%d",&l[i],&r[i]);
        v.push_back(l[i]);
        v.push_back(r[i]);
    }
    scanf("%d",&q);
    for (int i=0;i<q;++i) {
        scanf("%d%d",&x[i],&y[i]);
        v.push_back(x[i]);
        v.push_back(y[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for (int i=0;i<n;++i) {
        ++st[0][lower_bound(v.begin(),v.end(),l[i])-v.begin()+1];
        --st[0][lower_bound(v.begin(),v.end(),r[i])-v.begin()+2];
    }
    init();
    for (int i=0;i<q;++i) {
        tx=lower_bound(v.begin(),v.end(),x[i])-v.begin()+1;
        ty=lower_bound(v.begin(),v.end(),y[i])-v.begin()+1;
        printf("%d\n",query(tx,ty));
    }
    return 0;
}