SGU 507 Treediff

这个题目  其实可以暴力  用两个 set 合并； 每次放进去一个元素只要找到这个元素第一个比他大的元素和最后一个比他小的元素；然后更新最优值；

证明为什么不会超时；  假如最后集合的小的为 S1，大的集合为S2； 这样将 S1合并到S2； 这样S2 》= S1×2；合并 t 次后 大小就成了 S1×（2^t); 但是集合大小只有N 个元素  也就是说  N 》= S1×（2^t ）; 但S1等于1时 也就是说一个元素最多合并t 《= log(N) 次；

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
using namespace std;

struct date{
   int v,next;
}edge[51234]; int N,M,total,head[51234];
void add_edge( int u,int v ){
    edge[total].v = v;
    edge[total].next = head[u];
    head[u] = total++;
}
inline int min(int a,int b){return a<b?a:b;}
inline int max(int a,int b){return a>b?a:b;}
set<int>s[51234]; int res[51234]; int tab[51234];
int dfs( int sta )
{
    bool fell = false; bool fall = false; int ans = 2147483647;
    for( int i = head[sta]; i != -1; i = edge[i].next )
    {
        int v = edge[i].v;  ans = min( ans,dfs( v ) );   set<int>::iterator it,now;
        if( !fell ){ tab[sta] = tab[v]; res[sta] = 2147483647; fell = true; continue; }
        int len1 = s[tab[sta]].size(); int len2 = s[tab[v]].size();
        if( len1 > len2 ){
            for( it = s[tab[v]].begin(); it != s[tab[v]].end(); it++ )
            {
                 now = s[tab[sta]].lower_bound(*it);
                 if( now != s[tab[sta]].end() ){ ans = min( ans,*now-*it ); }
                 if( now != s[tab[sta]].begin() ){ now--;ans = min( ans,*it-*now ); }
                 if( s[tab[sta]].find(*it) == s[tab[sta]].end() ) s[tab[sta]].insert(*it);
            }
        }else {
            for( it = s[tab[sta]].begin(); it != s[tab[sta]].end(); it++ )
            {
                 now = s[tab[v]].lower_bound(*it);
                 if( now != s[tab[v]].end() ){ ans = min( ans,*now-*it ); }
                 if( now != s[tab[v]].begin() ){ ans = min( ans,*it-*(--now) ); }
                 if( s[tab[v]].find(*it) == s[tab[v]].end() )s[tab[v]].insert(*it);
            }
            tab[sta] = tab[v];
        }
    }
    res[sta] = ans; return ans;
}
int main( )
{
    while( scanf("%d%d",&N,&M) != EOF )
    {
        for( int i = 0; i <= N; i++ )s[i].clear();
        for( int i = 0; i <= N; i++ )tab[i] = i;
          total = 0; int k = N-M+1; memset( head,-1,sizeof(head) );
        for( int i = 2; i <= N ;i++ ){
            int v; scanf("%d",&v);
            add_edge( v,i );
        }
        for( int i = 1; i <= M; i++ ){
            int num; scanf("%d",&num); s[k++].insert(num);
        }
        dfs( 1 );
        printf("%d",res[1]);
        for( int i = 2; i <= N-M; i++ )
        printf(" %d",res[i]);
        puts("");
    }
    return 0;
}