dsu on tree(无讲解)

Posted 2021-01-22 suika

CF741D. Arpa’s letter-marked tree and Mehrdad’s Dokhtar-kosh paths

分析:

• 最多有一个字符出现奇数次
• 维护某个状态下深度的最大值，注意是全局深度
• 写成非递归形式方便理解

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
using namespace std;
#define N 500050
#define M (1<<22)
#define inf 0x3f3f3f3f
int head[N],to[N<<1],nxt[N<<1],cnt,n,val[N<<1];
int mx[M],siz[N],ans[N],son[N];
int dfn[N],idf[N],enp[N];
int nowans,dep[N],len[N];
inline void add(int u,int v,int w) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
}
void df1(int x,int y) {
    int i;
    siz[x]=1; len[x]=len[y]+1;
    dfn[x]=++dfn[0]; idf[dfn[0]]=x;
    for(i=head[x];i;i=nxt[i]) if(to[i]!=y) {
        dep[to[i]]=dep[x]^(1<<val[i]);
        df1(to[i],x);
        siz[x]+=siz[to[i]];
        if(siz[to[i]]>siz[son[x]]) son[x]=to[i];
    }
    enp[x]=dfn[0];
}
void df2(int x,int y,int opt) {
    int i;
    for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&to[i]!=son[x]) {
        df2(to[i],x,0); ans[x]=max(ans[x],ans[to[i]]);
    }
    if(son[x]) {
        df2(son[x],x,1);
        ans[x]=max(ans[x],ans[son[x]]);
        nowans=max(nowans,mx[dep[x]]-len[x]);
        for(i=0;i<22;i++) nowans=max(nowans,mx[dep[x]^(1<<i)]-len[x]);
    }
    mx[dep[x]]=max(mx[dep[x]],len[x]);
    for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&to[i]!=son[x]) {
        int t=to[i],j,p,k;
        for(j=dfn[t];j<=enp[t];j++) {
            p=idf[j];
            nowans=max(nowans,mx[dep[p]]+len[p]-len[x]*2);
            for(k=0;k<22;k++) nowans=max(nowans,mx[ dep[p]^(1<<k) ]+len[p]-len[x]*2);
        }
        for(j=dfn[t];j<=enp[t];j++) {
            p=idf[j];
            mx[dep[p]]=max(mx[dep[p]],len[p]);
        }
    }
    ans[x]=max(ans[x],nowans);
    if(!opt) {
        nowans=-inf;
        for(i=dfn[x];i<=enp[x];i++) mx[dep[idf[i]]]=-inf;
    }
}
int main() {
    memset(mx,0xc0,sizeof(mx));
    scanf("%d",&n);
    int i,x; 
    char op[4];
    for(i=2;i<=n;i++) {
        scanf("%d%s",&x,op);
        add(x,i,op[0]-'a'); add(i,x,op[0]-'a');
    }
    df1(1,0);
    df2(1,0,1);
    for(i=1;i<=n;i++) printf("%d ",ans[i]);
}

BZOJ5457: 城市

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
using namespace std;
#define N 400050
int head[N],to[N<<1],nxt[N<<1],n,m,cnt;
int a[N],b[N],dfn[N],idf[N],fa[N],enp[N],son[N],siz[N],top[N];
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
char buf[100000],*p1,*p2;
inline int rd() {
    int x=0;char c=nc();
    while(c<48)c=nc();
    while(c>47)x=((x+(x<<2))<<1)+(c^48),c=nc();
    return x;
}
inline void add(int u,int v) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void df1(int x,int y) {
    int i; fa[x]=y;
    siz[x]=1; dfn[x]=++dfn[0]; idf[dfn[0]]=x;
    for(i=head[x];i;i=nxt[i]) if(to[i]!=y) {
        df1(to[i],x);
        siz[x]+=siz[to[i]];
        if(siz[to[i]]>siz[son[x]]) son[x]=to[i];
    }
    enp[x]=dfn[0];
}
int nowans,ans[N],c[N],tot[N];
void df2(int x) {
    int i,j,t,lim;
    for(i=head[x];i;i=nxt[i]) if(to[i]!=son[x]&&to[i]!=fa[x]) {
        df2(to[i]);
    }
    if(son[x]) df2(son[x]);
    c[a[x]]+=b[x];
    if(c[nowans]<c[a[x]]||(c[nowans]==c[a[x]]&&nowans>a[x])) {
        nowans=a[x];
    }
    for(i=head[x];i;i=nxt[i]) if(to[i]!=son[x]&&to[i]!=fa[x]) {
        t=to[i],lim=enp[t];
        for(j=dfn[t];j<=lim;j++) {
            int p=idf[j];
            c[a[p]]+=b[p];
    if(c[nowans]<c[a[p]]||(c[nowans]==c[a[p]]&&nowans>a[p])) {
        nowans=a[p];
    }
        }
    }
    ans[x]=nowans; tot[x]=c[nowans];
    if(son[fa[x]]!=x) {nowans=0;for(lim=enp[x],i=dfn[x];i<=lim;i++) c[a[idf[i]]]=0;}
}
char pbuf[4000000],*pp=pbuf;
int sta[30],tp;
void write(int x) {
    if(x<0) *pp++='-',x=-x;
    do {sta[++tp]=x%10,x/=10;}while(x);
    while(tp)*pp++=sta[tp--]+'0';
}
int main() {
    n=rd(),m=rd();
    int i,x,y;
    for(i=1;i<n;i++) x=rd(),y=rd(),add(x,y),add(y,x);
    for(i=1;i<=n;i++) a[i]=rd(),b[i]=rd();
    df1(1,0); df2(son[0]=1);
    for(i=1;i<=n;i++) {
        write(ans[i]); *pp++=' '; write(tot[i]); *pp++='
';
    }
    fwrite(pbuf,1,pp-pbuf,stdout);
}