bzoj 3681: Arietta 主席树优化建图+网络流+dsu on tree

题意

所有的 n 个音符形成一棵由音符 C ( 1 号节点) 构成的有根树,每一个音符有一个音高 Hi 。
Arietta 有 m 个力度,第 i 个力度能弹出 Di 节点的子树中,音高在 [Li,Ri] 中的任意一个音符。
为了乐曲的和谐,Arietta 最多会弹奏第 i 个力度 Ti 次。
Arietta 想知道她最多能弹出多少个音符。
1 ≤ n, m ≤ 10000 。对于所有数据,1 ≤ Hi , Ti , Pi ≤ n, 1 ≤ Li ≤ Ri ≤ n 。

分析

这题一看就知道肯定是网络流没跑了，现在的问题是如何用数据结构来优化建图。
一开始想到了什么dfs序主席树啊树套树啊之类的，发现貌似都不行，于是就不会了。。。

Orz题解：
首先考虑一个点数换边数的做法：对每一棵子树都建一棵线段树，然后优化建图跑网络流。
这样显然是过不了的，但可以考虑用dsu on tree来优化这个做法。
我们先把整棵树轻重链剖分一下，设当前点为x，把x的重儿子的线段树赋为x的线段树，也就是可持久化。然后再暴力把x的所有轻儿子的子树和x本身都加进来，就得到了x子树的线段树。
复杂度的话，因为每个点到根的路径上，重链数量不超过logn条，所以每个点最多被加入logn次，于是点数就为$O(nlog^2n)$，同样的，边数也为$O(nlog^2n)$，于是就可以愉快的跑网络流啦。
注意主席树在加入新节点的时候要把该区间的新节点往原来的节点连边。

一次AC是真的开心！！！

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=10005;
const int M=N*130;
const int inf=0x3f3f3f3f;

int n,m,last[N],cnt,sz,h[N],size[N],mn[N],mx[N],dfn[N],root[N];
struct edge{int to,next;}e[N];
struct tree{int l,r;}t[M];

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;
}

namespace flow
{
    int cnt=1,s,t,last[M],cur[M],ans,dis[M];
    struct edge{int to,c,next;}e[N*300];
    queue<int> que;

    void addedge(int u,int v,int c)
    {
        e[++cnt].to=v;e[cnt].c=c;e[cnt].next=last[u];last[u]=cnt;
        e[++cnt].to=u;e[cnt].c=0;e[cnt].next=last[v];last[v]=cnt;
    }

    bool bfs()
    {
        for (int i=s;i<=t;i++) dis[i]=0;
        while (!que.empty()) que.pop();
        que.push(s);dis[s]=1;
        while (!que.empty())
        {
            int u=que.front();que.pop();
            for (int i=last[u];i;i=e[i].next)
                if (e[i].c&&!dis[e[i].to])
                {
                    dis[e[i].to]=dis[u]+1;
                    if (e[i].to==t) return 1;
                    que.push(e[i].to);
                }
        }
        return 0;
    }

    int dfs(int x,int maxf)
    {
        if (x==t||!maxf) return maxf;
        int ret=0;
        for (int &i=cur[x];i;i=e[i].next)
            if (e[i].c&&dis[e[i].to]==dis[x]+1)
            {
                int f=dfs(e[i].to,min(e[i].c,maxf-ret));
                e[i].c-=f;
                e[i^1].c+=f;
                ret+=f;
                if (maxf==ret) break;
            }
        return ret;
    }

    void dinic()
    {
        while (bfs())
        {
            for (int i=s;i<=t;i++) cur[i]=last[i];
            ans+=dfs(s,inf);
        }
    }
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void dfs1(int x)
{
    mn[x]=++sz;dfn[sz]=x;size[x]=1;
    for (int i=last[x];i;i=e[i].next) dfs1(e[i].to),size[x]+=size[e[i].to];
    mx[x]=sz;
}

void ins(int &d,int p,int l,int r,int x,int y)
{
    d=++sz;t[d]=t[p];flow::addedge(d+n,p+n,inf);
    if (l==r)
    {
        flow::addedge(d+n,y,inf);
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,t[p].l,l,mid,x,y),flow::addedge(d+n,t[d].l+n,inf);
    else ins(t[d].r,t[p].r,mid+1,r,x,y),flow::addedge(d+n,t[d].r+n,inf);
}

void build(int d,int l,int r,int x,int y,int z)
{
    if (!d||x>y) return;
    if (l==x&&r==y)
    {
        flow::addedge(z,d+n,inf);
        return;
    }
    int mid=(l+r)/2;
    build(t[d].l,l,mid,x,min(y,mid),z);
    build(t[d].r,mid+1,r,max(x,mid+1),y,z);
}

void add(int x,int y)
{
    for (int i=mn[x];i<=mx[x];i++) ins(root[y],root[y],1,n,h[dfn[i]],dfn[i]);
}

void dfs2(int x)
{
    int k=0;
    for (int i=last[x];i;i=e[i].next) k=size[e[i].to]>size[k]?e[i].to:k;
    for (int i=last[x];i;i=e[i].next) if (e[i].to!=k) dfs2(e[i].to);
    if (k) dfs2(k),root[x]=root[k];
    for (int i=last[x];i;i=e[i].next) if (e[i].to!=k) add(e[i].to,x);
    ins(root[x],root[x],1,n,h[x],x);
}

int main()
{
    n=read();m=read();
    for (int i=2;i<=n;i++)
    {
        int x=read();
        addedge(x,i);
    }
    for (int i=1;i<=n;i++) h[i]=read();
    dfs1(1);
    dfs2(1);
    flow::s=0;flow::t=n+m+sz+1;
    for (int i=1;i<=m;i++)
    {
        int l=read(),r=read(),x=read(),t=read();
        flow::addedge(flow::s,n+sz+i,t);
        build(root[x],1,n,l,r,n+sz+i);
    }
    for (int i=1;i<=n;i++) flow::addedge(i,flow::t,1);
    flow::dinic();
    printf("%d",flow::ans);
    return 0;
}