BZOJ 3681: Arietta

这题跟前一题差不多

同样是用主席树来优化建图
因为要保存每个点的信息 所以孩子合并到父亲的时候要新开一个节点
把这两个点并起来 总的来算 其实也就多了一倍嘛。。
一开始的打法真的蠢QAQ

#include<bits/stdc++.h>
using namespace std;
const int N=10005,inf=1e9+7;
inline int read(){
    char ch=getchar(); int x=0,f=1;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0'; ch=getchar();}
    return x*f;
}
struct E{int y,c,next;}e[N*100],a[N]; int len,first[N*50],fir[N];
void ins(int x,int y,int c){
    e[++len]=(E){y,c,first[x]},first[x]=len;
    e[++len]=(E){x,0,first[y]},first[y]=len;
}
int h[N*50],t,tot; queue<int>q;
bool bfs(){
    for(int i=0;i<=tot;i++)h[i]=0;
    q.push(0); h[0]=1;
    while(!q.empty()){
        int x=q.front(); q.pop();
        for(int k=first[x];k;k=e[k].next){
            int y=e[k].y;
            if(!h[y] && e[k].c){h[y]=h[x]+1; q.push(y);}
        }
    }
    return h[t]>0;
}
int dfs(int x,int f){
    if(x==t)return f;
    int de=0;
    for(int k=first[x];k;k=e[k].next){
        int y=e[k].y;
        if(h[y]==h[x]+1){
            int sf=dfs(y,min(f-de,e[k].c));
            e[k].c-=sf,e[k^1].c+=sf,de+=sf;
            if(de==f)return f;
        }
    }
    if(!de)h[x]=0;
    return de;
}
int rt[N],lc[N*50],rc[N*50],n;
void link(int &x,int l,int r,int u){
    x=++tot;
    if(l==r){
        ins(0,x,1); return;
    }
    int mid=(l+r)>>1;
    if(u<=mid)link(lc[x],l,mid,u),ins(lc[x],x,inf);
    else link(rc[x],mid+1,r,u),ins(rc[x],x,inf);
}
int merge(int x,int y,int l,int r){
    if(x*y==0)return x+y;
    int o=++tot;
    if(l==r)ins(x,o,inf),ins(y,o,inf);
    else{
        int mid=(l+r)>>1;
        if(lc[x]+lc[y])ins(lc[o]=merge(lc[x],lc[y],l,mid),o,inf);
        if(rc[x]+rc[y])ins(rc[o]=merge(rc[x],rc[y],mid+1,r),o,inf);
    }
    return o;
}
void bulid(int x){
    for(int k=fir[x];k;k=a[k].next){
        int y=a[k].y;
        bulid(y); rt[x]=merge(rt[x],rt[y],1,n);
    }
}
int ql,qr;
void add(int x,int l,int r){
    if(ql>r || qr<l)return;
    if(ql<=l && r<=qr){
        ins(x,tot,inf); return;
    }
    int mid=(l+r)>>1;
    if(lc[x])add(lc[x],l,mid);
    if(rc[x])add(rc[x],mid+1,r);
}
int main()
{
    n=read(); int m=read(),i,x,y,c;
    for(i=2;i<=n;i++)
        x=read(),a[i]=(E){i,0,fir[x]},fir[x]=i;
    len=tot=t=1;
    for(i=1;i<=n;i++)
        x=read(),link(rt[i],1,n,x);
    bulid(1);
    for(i=1;i<=m;i++){
        ql=read(),qr=read(),x=read(),c=read();
        tot++; add(rt[x],1,n); ins(tot,t,c);
    }
    int ans=0;
    while(bfs())
        ans+=dfs(0,inf);
    printf("%d\n",ans);
    return 0;
}