[JZOJ100019]A

题目大意

求一颗树有多少条不同的路径，使得路径长度>1且不存在j!=k满足j是k倍数。

瞎做

有n log n个限制某两个点不能出现在一条路径中。
可以转化成某个区间的起点不能以某个区间为终点。
然后扫描线用线段树维护。
这个是既包含插入又包含删除的覆盖问题。
因为操作具体对称性所以可以标记可持久化。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=150000+10,maxm=maxn*200;
int h[maxn],go[maxn*2],next[maxn*2];
int h2[maxn],L[maxm],R[maxm],fx[maxm],n2[maxm];
int dfn[maxn],nfd[maxn],size[maxn],dep[maxn],zjy[maxn];
int tree[maxn*4],bz[maxn*4];
int f[maxn][25];
int i,j,k,l,t,n,m,tot,top,cnt;
ll ans;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    dep[x]=dep[y]+1;
    f[x][0]=y;
    dfn[x]=++top;
    nfd[top]=x;
    size[x]=1;
    int t=h[x];
    while (t){
        if (go[t]!=y){
            dfs(go[t],x);
            size[x]+=size[go[t]];
        }
        t=next[t];
    }
}
int find(int x,int y){
    int j=zjy[dep[x]];
    while (j>=0){
        if (y>=(1<<j)){
            x=f[x][j];
            y-=(1<<j);
        }
        j--;
    }
    return x;
}
void add2(int x,int l,int r,int y){
    L[++tot]=l;
    R[tot]=r;
    fx[tot]=y;
    n2[tot]=h2[x];
    h2[x]=tot;
}
void cr(int l1,int r1,int l2,int r2){
    if (l1>r1||l2>r2) return;
    add2(l1,l2,r2,1);
    add2(r1+1,l2,r2,-1);
}
void ins(int x,int y){
    if (x==y){
        cr(dfn[x],dfn[x],dfn[x],dfn[x]);
        /*add2(dfn[x],dfn[y],dfn[y],1);
        add2(dfn[x]+1,dfn[y],dfn[y],-1);*/
        return;
    }
    if (dfn[x]>dfn[y]) swap(x,y);
    if (dfn[x]+size[x]>=dfn[y]+size[y]){
        int z=find(y,dep[y]-dep[x]-1);
        cr(dfn[y],dfn[y]+size[y]-1,1,dfn[z]-1);
        cr(dfn[y],dfn[y]+size[y]-1,dfn[z]+size[z],n);
        cr(1,dfn[z]-1,dfn[y],dfn[y]+size[y]-1);
        cr(dfn[z]+size[z],n,dfn[y],dfn[y]+size[y]-1);
        /*add2(dfn[x],1,dfn[z]-1,1);
        add2(dfn[x],dfn[z]+size[z],n,1);
        add2(dfn[x]+size[x],1,dfn[z]-1,-1);
        add2(dfn[x]+size[x],dfn[z]+size[z],n,-1);*/
    }
    else{
        cr(dfn[x],dfn[x]+size[x]-1,dfn[y],dfn[y]+size[y]-1);
        swap(x,y);
        cr(dfn[x],dfn[x]+size[x]-1,dfn[y],dfn[y]+size[y]-1);
        /*add2(dfn[x],dfn[y],dfn[y]+size[y]-1,1);
        add2(dfn[x]+size[x],dfn[y],dfn[y]+size[y]-1,-1);*/
    }
}
void update(int p,int l,int r){
    if (bz[p]) tree[p]=r-l+1;
    else tree[p]=tree[p*2]+tree[p*2+1];
}
void change(int p,int l,int r,int a,int b,int v){
    if (a>b) return;
    if (l==a&&r==b){
        bz[p]+=v;
        update(p,l,r);
        return;
    }
    int mid=(l+r)/2;
    if (b<=mid) change(p*2,l,mid,a,b,v);
    else if (a>mid) change(p*2+1,mid+1,r,a,b,v);
    else change(p*2,l,mid,a,mid,v),change(p*2+1,mid+1,r,mid+1,b,v);
    update(p,l,r);
}
int main(){
    freopen("a.in","r",stdin);freopen("a.out","w",stdout);
    n=read();
    fo(i,1,n-1){
        j=read();k=read();
        add(j,k);add(k,j);
    }
    dfs(1,0);
    fo(i,1,n) zjy[i]=floor(log(i)/log(2));
    fo(j,1,zjy[n])
        fo(i,1,n)
            f[i][j]=f[f[i][j-1]][j-1];
    tot=0;
    fo(i,1,n){
        ins(i,i);
        fo(j,2,n/i) ins(i,i*j);
    }
    ans=0;
    fo(i,1,n){
        j=i;
        t=h2[j];
        while (t){
            if (fx[t]==-1) change(1,1,n,L[t],R[t],-1);
            t=n2[t];
        }
        t=h2[j];
        while (t){
            if (fx[t]==1) change(1,1,n,L[t],R[t],1);
            t=n2[t];
        }
        ans+=(ll)(n-tree[1]);
    }
    ans/=2;
    printf("%lld\n",ans);
}