cf369 E. Valera and Queries

链接

点击跳转

题解

为了不重复统计,我可以把每条线段在其包含的最左边的查询点处统计

比如先统计左端点在 [ 1 , p 1 ] [1,p_1] [1,p1],右端点在 [ p 1 , + ∞ ] [p_1,+\infin] [p1,+]
然后统计左端点在 ( p 1 , p 2 ] (p_1,p_2] (p1,p2],右端点在 [ p 2 , + ∞ ] [p_2,+\infin] [p2,+]
依次类推

主席树即可

代码

#include 
#include 
#include 
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
struct PersistenceSegmentTree
{
    int root[maxn<<4], rtot, tot, L[maxn<<4], R[maxn<<4], ch[maxn<<4][2], v[maxn<<4];
    int copy(int o)
    {
        tot++;
        L[tot]=L[o], R[tot]=R[o], ch[tot][0]=ch[o][0], ch[tot][1]=ch[o][1], v[tot]=v[o];
        return tot;
    }
    int New()
    {
        tot++;
        L[tot]=R[tot]=ch[tot][0]=ch[tot][1]=v[tot]=0;
        return tot;
    }
    void build(int o, int l, int r)
    {
        int mid(l+r>>1);
        L[o]=l, R[o]=r;
        if(l==r)return;
        ch[o][0]=New(), ch[o][1]=New();
        build(ch[o][0],l,mid);
        build(ch[o][1],mid+1,r);
    }
    void init(int l, int r)
    {
        rtot=0, tot=0;
        build(root[0]=New(),l,r);
    }
    int insert(int o, int pos, int add)
    {
        int x=copy(o), mid=L[o]+R[o]>>1;
        v[x]+=add;
        if(L[o]==R[o])return x;
        if(pos<=mid)ch[x][0]=insert(ch[o][0],pos,add);
        else ch[x][1]=insert(ch[o][1],pos,add);
        return x;
    }
    void ins(int pos, int add=1)
    {
        root[rtot+1]=insert(root[rtot],pos,add);
        rtot++;
    }
    int qsum(int o1, int o2, int l, int r)
    {
        int ans=0, mid=L[o1]+R[o1]>>1;
        if(l<=L[o1] and r>=R[o1])return v[o2]-v[o1];
        if(l<=mid)ans+=qsum(ch[o1][0],ch[o2][0],l,r);
        if(r>mid)ans+=qsum(ch[o1][1],ch[o2][1],l,r);
        return ans;
    }
    int kth(int o1, int o2, int k)
    {
        if(L[o1]==R[o1])return L[o1];
        int t=v[ch[o2][0]]-v[ch[o1][0]];
        if(t>=k)return kth(ch[o1][0],ch[o2][0],k);
        else return kth(ch[o1][1],ch[o2][1],k-t);
    }
}zx;
int n, m, l[maxn], r[maxn], p[maxn], cnt, b[maxn], id[maxn];
int main()
{
    int i, j, ans=0;
    n=read(), m=read();
    rep(i,n)
    {
        l[i]=read(), r[i]=read();
        b[l[i]]++;
        id[i]=i;
    }
    rep(i,1000000)b[i]+=b[i-1];
    sort(id+1,id+n+1,[](int x, int y){return l[x]<l[y];});
    zx.init(1,1000000);
    rep(i,n)zx.ins(r[id[i]]);
    while(m--)
    {
        cnt=read();
        rep(i,cnt)p[i]=read();
        ans=0;
        rep(i,cnt)
        {
            ans+=zx.qsum(zx.root[b[p[i-1]]],zx.root[b[p[i]]],p[i],1000000);
        }
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(#,主席树)