BZOJ2783: [JLOI2012]树

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=2783

题解：因为深度必须是递增的，所以就是从上到下的一条路径。

         令d[i]表示i到根的路径，那么如果从x出发的满足题意的终点y必然有d[y]-d[fa[x]]=s

         查询某个子树内某个值的出现次数？主席树！！！

         d[i]太大？离散化就行。我省事用了map。

代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 100000+5

#define maxm 2000000+5

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go)

#define for5(n,m) for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)

#define mod 1000000007

#define mid ((l+r)>>1)

#define lch k<<1,l,mid

#define rch k<<1|1,mid+1,r

using namespace std;

inline 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=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,m,ti,cnt,tot,head[maxn],a[maxn],id[maxn],pos[maxn][2],sum,fa[maxn],rt[maxn];
int s[maxm],ls[maxm],rs[maxm];
struct edge{int go,next;}e[maxn];
map<int,int>mp;
inline void add(int x,int y)
{
    e[++tot]=(edge){y,head[x]};head[x]=tot;
}
inline void dfs(int x)
{
    pos[x][0]=++ti;id[ti]=x;
    for4(i,x)if(!pos[y][0])
    {
        fa[y]=x;a[y]+=a[x];
        dfs(y);
    }
    pos[x][1]=ti;
}
inline void update(int l,int r,int x,int &y,int z)
{
    y=++sum;
    s[y]=s[x]+1;
    if(l==r)return;
    ls[y]=ls[x];rs[y]=rs[x];
    if(z<=mid)update(l,mid,ls[x],ls[y],z);else update(mid+1,r,rs[x],rs[y],z);
}
inline int query(int l,int r,int x,int y,int z)
{
    if(l==r)return s[y]-s[x];
    return z<=mid?query(l,mid,ls[x],ls[y],z):query(mid+1,r,rs[x],rs[y],z);
}

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();m=read();
    for1(i,n)a[i]=read();
    for1(i,n-1){int x=read(),y=read();add(x,y);}
    dfs(1);
    for1(i,n)if(!mp[a[i]])mp[a[i]]=++cnt;
    for1(i,n)update(1,cnt,rt[i-1],rt[i],mp[a[id[i]]]);
    ll ans=0;
    for1(i,n)
    {
        int t=mp[a[fa[i]]+m];
        if(t)ans+=query(1,cnt,rt[pos[i][0]-1],rt[pos[i][1]],t);
    }
    cout<<ans<<endl;

    return 0;

}  

看着status里面一堆人代码超短有种不详的预感。。。

UPD：T_T我的做法好无脑。。。

          还有一种比较好想到的做法是倍增，对于一个终点倍增+二分出满足题意的起点。戳这里：http://jiruyi910387714.is-programmer.com/posts/40545.html

          还有黄巨大的set做法，真是orz http://hzwer.com/4237.html