BZOJ 4919 大根堆

Problem

BZOJ

Solution

一个树上lis问题。处理方式很类似于序列上的lis问题的nlogn解法。
每个节点维护一个set，set就类似于nlogn中的维护的单调序列。
将节点x的子树启发式合并一下，然后找到第一个大于等于val[x]的值修改为val[x]即可。
最后的答案就是s[1].size()。

Code

#include 
#include 
#include 
using namespace std;
const int maxn=200010;
struct data{int v,nxt;}edge[maxn<<1];
int n,m,p,a[maxn],head[maxn];
multiset<int> s[maxn];
multiset<int>::iterator it;
inline void insert(int u,int v){edge[++p]=(data){v,head[u]};head[u]=p;}
void merge(int x,int y)
{
    if(s[x].size()>s[y].size()) swap(s[x],s[y]);
    it=s[x].begin();
    while(it!=s[x].end()){s[y].insert(*it);it++;}
    s[x].clear();
}
void dfs(int x)
{
    for(int i=head[x];i;i=edge[i].nxt)
    {
        dfs(edge[i].v);
        merge(edge[i].v,x);
    }
    it=s[x].lower_bound(a[x]);
    if(it!=s[x].end()) s[x].erase(it);
    s[x].insert(a[x]);
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    scanf("%d",&n);
    for(int i=1,x;i<=n;i++)
    {
        scanf("%d%d",&a[i],&x);
        if(x) insert(x,i);
    }
    dfs(1);
    printf("%d\n",(int)s[1].size());
    return 0;
}