[JZOJ100019]A--dfn序+扫描线

[JZOJ100019]A--dfn序+扫描线

题目链接

太懒了自行搜索

分析

这道题查了一个下午的错,真的心态崩了

不过这道题确实妙啊

类比于喝喝喝,我们发现任何一条覆盖了非法路径的路径一定不合法,假如非法路径为\((x,kx)\),设\(u\)为\(dfn\)序中较小的那个点,\(v\)为较大的那个;假如覆盖了路径\((u,v)\)的路径为\((a,b)(dfn_a<dfn_b)\);

设数组\(ed[x]\)为\(x\)的子树中\(dfs\)最大的那个点的\(dfs\)序,显然这可以和\(dfs\)序在一遍\(dfs\)中求出

我们还需要知道:\(x\) 的子树中\(dfs\)序是连续的一段

考虑两种情况:

• \(u\)是\(lca(u,v)\),分析发现这也有两种情况
  先设\(g\)为路径\((u,v)\)上距\(u\)最近的那个点

  • Case#1
    \(dfn[v]<=dfn[a]<=ed[v]\) , \(dfn[b]>ed[g]\)
    

  • Case#2

  \(dfn[v]<=dfn[b]<=ed[v]\) , \(dfn[g]>dfn[a]\)
  

• \(u\)不是\(lca(u,v)\)

  这个只有一种情况,\(a,b\)分别在\(u,v\)子树中,即

  Case#3

  \(dfn[u]<=dfn[a]<=ed[u]<=dfn[v]<=dfn[b]<=ed[v]\)

我们把\((a,b)\)看成一个有序数对的话,发现其实每一种\(case\)都围成了一个矩形,我们只需要求出矩形面积的并之后即可得出所有不合法的路径,用总路径数\(n*(n-1)/2\)减去不合法即所求

求矩形面积的并使用线段树的扫描线法,但是发现一个点我们也要算进贡献,并非严格意义的二维图形；

因为值域很小\((1e5)\),学会了一种船新操作,我们不排序,将纵坐标相同的丢入一个容器,直接从小到大钦定纵坐标.不断取出对应容器内的线段加入,然后计算剩余的点数即可(纵坐标已钦定)

一道图论题转化成了数据结构,也是精妙啊

代码

/*
  code by RyeCatcher
*/
inline char gc(){
    static char buf[SIZE],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,SIZE,stdin),p1==p2)?EOF:*p1++;
}
template <class T>inline void read(T &x){
    x=0;int ne=0;char c;
    while((c=gc())>'9'||c<'0')ne=c=='-';x=c-48;
    while((c=gc())>='0'&&c<='9')x=(x<<3)+(x<<1)+c-48;x=ne?-x:x;return ;
}
const int maxn=500005;
const int inf=0x7fffffff;
int n;
struct Edge{
    int ne,to;
}edge[maxn<<1];
int h[maxn],num_edge=1;
int fa[maxn][17];
inline void add_edge(int f,int to){
    edge[++num_edge].ne=h[f];
    edge[num_edge].to=to;
    h[f]=num_edge;
}
int dfn[maxn],ed[maxn],dep[maxn],tot=0;
void dfs(int now){
    int v;dfn[now]=++tot;
    for(ri i=h[now];i;i=edge[i].ne){
        v=edge[i].to;
        if(v==fa[now][0]||dfn[v])continue;
        dep[v]=dep[now]+1,fa[v][0]=now;
        for(ri i=1;i<=16;i++)fa[v][i]=fa[fa[v][i-1]][i-1];
        dfs(v);
    }
    ed[now]=tot;
    return ;
}
int get_g(int x,int y){//x到y路径上距y最近 
    for(ri i=16;i>=0;i--){
        //printf("%d %d\n",fa[x][i],x);
        if(dep[fa[x][i]]>dep[y]){
            x=fa[x][i];
            //printf("%d\n",x);
        }
    }
    return x;
}
int L,R,dta;
struct Seg{
    int l,r,h,d;
    Seg(){l=r=h=d=0;}
    Seg(int _l,int _r,int _h,int _d){l=_l,r=_r,h=_h,d=_d;}
}seg[maxn<<4];
int poi=0;
ll sum[maxn<<2];
ll tag[maxn<<2];
vector <int> dd[maxn];
inline void modify(int now,int l,int r){
    if(tag[now]>0)sum[now]=(r-l+1);
    else sum[now]=sum[now<<1]+sum[now<<1|1];
}
void update(int now,int l,int r){
    if(L<=l&&r<=R){
        tag[now]+=dta;
        modify(now,l,r);
        return ;
    }
    int mid=(l+r)>>1;
    if(L<=mid)update(now<<1,l,mid);
    if(mid<R)update(now<<1|1,mid+1,r);
    modify(now,l,r);
    return ;
}
ll ans=0;
int main(){
    int x,y,ex,ey,g;
    FO(a);
    //freopen("a5.in","r",stdin);
    read(n);
    for(ri i=1;i<n;i++){
        read(x),read(y);
        add_edge(x,y);
        add_edge(y,x);
    }
    dep[1]=1,fa[1][0]=0;
    dfs(1);
    int p,q;
    for(ri i=1;i<=n;i++){
        for(ri j=i+i;j<=n;j+=i){
            p=i,q=j;
            if(dfn[p]<dfn[q])swap(p,q);
            x=dfn[p],y=dfn[q];
            ex=ed[p],ey=ed[q];
            if(x>=y&&x<=ey){
                g=get_g(p,q);
                dd[x].push_back(++poi);
                seg[poi]=Seg(1,dfn[g]-1,x,1);
                dd[ex+1].push_back(++poi);
                seg[poi]=Seg(1,dfn[g]-1,ex,-1);
                if(ed[g]==n)continue;
                dd[ed[g]+1].push_back(++poi);
                seg[poi]=Seg(x,ex,ed[g]+1,1);
                dd[n+1].push_back(++poi);
                seg[poi]=Seg(x,ex,n,-1);
            }
            else {
                dd[x].push_back(++poi);
                seg[poi]=Seg(y,ey,x,1);
                dd[ex+1].push_back(++poi);
                seg[poi]=Seg(y,ey,ex,-1);
            }
        }
    }
    for(ri i=1;i<=n;i++){
        for(ri j=0;j<dd[i].size();j++){
            x=dd[i][j];
            L=seg[x].l,R=seg[x].r,dta=seg[x].d;
            update(1,1,n);
        }
        ans+=sum[1];
    }
    printf("%lld\n",1ll*n*(n-1)/2-ans);
    return 0;
}

转载于:https://www.cnblogs.com/Rye-Catcher/p/9807676.html