UOJ_407_IOI2018狼人

Posted 2021-01-22 suika

http://uoj.ac/problem/407

分析:

• 分别建立最小/最大kruskal重构树。
• 每次询问给出的两个点能走到的部分分别对应两棵树(dfs)序的一段区间。
• 转化成判断矩形中是否有点。

代码:

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include "werewolf.h"
using namespace std;
#define M 400050
#define vi vector<int>
int n,m,q_times,fa[M],ln;
int dfn[2][M],enp[2][M],Ls[2][M],Rs[2][M],dfc,w[2][M];
int f[2][20][M],idf[2][M],root[M],ls[M*22],rs[M*22],siz[M*22],cnt;
char buf[100000],*p1,*p2;
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {
    int x=0; char s=nc();
    while(s<'0') s=nc();
    while(s>='0') x=(((x<<2)+x)<<1)+s-'0',s=nc();
    return x;
}
struct A {
    int a,b,c;
    bool operator < (const A &u) const {
        return c>u.c;
    }
}e[M];
int find(int x) {return fa[x]==x?x:fa[x]=find(fa[x]);}
void df1(int x,int o) {
    dfn[o][x]=++dfc;
    idf[o][dfc]=x;
    if(Ls[o][x]) df1(Ls[o][x],o);
    if(Rs[o][x]) df1(Rs[o][x],o);
    enp[o][x]=dfc;
}
void update(int l,int r,int x,int &p,int q) {
    p=++cnt; siz[p]=siz[q]+1; ls[p]=ls[q]; rs[p]=rs[q];
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(x<=mid) update(l,mid,x,ls[p],ls[q]);
    else update(mid+1,r,x,rs[p],rs[q]);
}
int query(int l,int r,int x,int y,int p,int q) {
    if(x<=l&&y>=r) return siz[p]-siz[q];
    int mid=(l+r)>>1,re=0;
    if(x<=mid) re+=query(l,mid,x,y,ls[p],ls[q]);
    if(y>mid) re+=query(mid+1,r,x,y,rs[p],rs[q]);
    return re;
}
vi check_validity(int n,vi EX,vi EY,vi QX,vi QY,vi QL,vi QR) {
    ln=n*2-1;
    m=EX.size();
    int i,tot=n;
    for(i=1;i<=m;i++) {
        e[i].a=EX[i-1]; e[i].b=EY[i-1];
        e[i].a++; e[i].b++;
        e[i].c=min(e[i].a,e[i].b);
    }
    sort(e+1,e+m+1);
    for(i=1;i<=ln;i++) fa[i]=i;
    for(i=1;i<=m;i++) {
        int dx=find(e[i].a),dy=find(e[i].b);
        if(dx!=dy) {
            fa[dx]=fa[dy]=++tot; Ls[0][tot]=dx; Rs[0][tot]=dy; w[0][tot]=e[i].c; f[0][0][dx]=f[0][0][dy]=tot;
        }
    }
    df1(tot,0); dfc=0;
    for(i=1;i<=ln;i++) fa[i]=i;
    tot=n;
    for(i=1;i<=m;i++) e[i].c=max(e[i].a,e[i].b);
    sort(e+1,e+m+1);
    for(i=m;i;i--) {
        int dx=find(e[i].a),dy=find(e[i].b);
        if(dx!=dy) {
            fa[dx]=fa[dy]=++tot; Ls[1][tot]=dx; Rs[1][tot]=dy; w[1][tot]=e[i].c; f[1][0][dx]=f[1][0][dy]=tot;
        }
    }
    df1(tot,1);
    int j,k;
    for(i=0;i<2;i++) {
        for(j=1;(1<<j)<=ln;j++) {
            for(k=1;k<=ln;k++) {
                f[i][j][k]=f[i][j-1][f[i][j-1][k]];
            }
        }
    }
    for(i=1;i<=dfc;i++) {
        root[i]=root[i-1];
        int x=idf[1][i];
        if(x<=n) {
            update(1,dfc,dfn[0][x],root[i],root[i]);
        }
    }
    vi ans;
    int x,y,l,r;
    q_times=QX.size();
    for(j=0;j<q_times;j++) {
        x=QX[j]; y=QY[j]; l=QL[j]; r=QR[j];
        x++;y++;l++;r++;
        for(i=19;i>=0;i--) {
            if(f[0][i][x]&&w[0][f[0][i][x]]>=l) x=f[0][i][x];
            if(f[1][i][y]&&w[1][f[1][i][y]]<=r) y=f[1][i][y];
        }
        int t=query(1,dfc,dfn[0][x],enp[0][x],root[enp[1][y]],root[dfn[1][y]-1]);
        ans.push_back(t>0);
    }
    return ans;
}