p4899 [IOI2018] werewolf 狼人

分析

我用的主席树维护qwq

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
struct node {
    int le,ri,sum;
};
node d[*];
int n,m,q,val[],rt[],cnt;
vector<int>v[];
struct Tree {
    int kk,dfn[],fin[],fa[],pa[][],T;
    vector<int>vv[];
    inline int sf(int x){return fa[x]==x?x:fa[x]=sf(fa[x]);}
    inline void dfs(int x){
      dfn[x]=++T;
      for(int i=;i<;i++)pa[x][i]=pa[pa[x][i-]][i-];
      for(int i=;i<vv[x].size();i++)dfs(vv[x][i]);
      fin[x]=T;
    }
    inline void work(){
      int i,j,k;
      for(i=;i<=n;i++)fa[i]=i;
      if(kk){
          for(i=n;i;i--)
            for(j=;j<v[i].size();j++)
              if(i<v[i][j]){
                k=sf(v[i][j]);
                if(k==i)continue;
                vv[i].push_back(k);
                fa[k]=pa[k][]=i;
            }
        dfs();
      }else {
          for(i=;i<=n;i++)
            for(j=;j<v[i].size();j++)
              if(i>v[i][j]){
                k=sf(v[i][j]);
                if(k==i)continue;
                vv[i].push_back(k);
                fa[k]=pa[k][]=i;
            }
        dfs(n);
      }
    }
    inline int go(int x,int len){
      int i,j,k;
      for(i=;i>=;i--)
        if(pa[x][i]&&((kk&&pa[x][i]>=len)||(!kk&&pa[x][i]<=len)))
          x=pa[x][i];
      return x;
    }
};
Tree A,B;
inline void update(int &x,int y,int le,int ri,int k){
    d[++cnt]=d[y];
    d[cnt].sum++;
    x=cnt;
    if(le==ri)return;
    int mid=(le+ri)>>;
    if(mid>=k)update(d[x].le,d[y].le,le,mid,k);
      else update(d[x].ri,d[y].ri,mid+,ri,k);
}
inline int que(int a,int b,int le,int ri,int x,int y){
    if(le>=x&&ri<=y)return d[a].sum-d[b].sum;
    int mid=(le+ri)>>,Ans=;
    if(mid>=x)Ans+=que(d[a].le,d[b].le,le,mid,x,y);
    if(mid<y)Ans+=que(d[a].ri,d[b].ri,mid+,ri,x,y);
    return Ans;
}
int main(){
    int i,j,k,x,y;
    scanf("%d%d%d",&n,&m,&q);
    for(i=;i<=m;i++){
      scanf("%d%d",&x,&y);
      x++,y++;
      v[x].push_back(y);
      v[y].push_back(x);
    }
    A.kk=;A.work();B.work();
    for(i=;i<=n;i++)val[A.dfn[i]]=B.dfn[i];
    for(i=;i<=n;i++)update(rt[i],rt[i-],,n,val[i]);
    for(i=;i<=q;i++){
      int s,t,le,ri;
      scanf("%d%d%d%d",&s,&t,&le,&ri);
      s++,t++,le++,ri++;
      s=A.go(s,le),t=B.go(t,ri);
      if(que(rt[A.fin[s]],rt[A.dfn[s]-],,n,B.dfn[t],B.fin[t]))puts("");
        else puts("");
    }
    return ;
}