hdu 4366 Successor(线段树)

http://acm.hdu.edu.cn/showproblem.php?pid=4366

给定后继关系，寻找后继者满足能力值大于被继承者中忠诚值最大的。

先将树形结构变为线性结构，用邻接表+dfs处理。然后对能力值大到小排序，保证查询时已经插入的一定是能力值大于该查询的值。

线段树功能转化为求解区间最值。

本来1Y的代码，结果loy和blt的输入搞反了！试了好多数据还没试出来！郁闷，这两天写的代码好多这种脑残错。。。

code:

#include <cstdlib>
#include <cctype>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
#include <sstream>
#include <set>
#include <queue>
#include <stack>
#include <fstream>
#include <iomanip>
#include <bitset>
#include <list>
#include <ctime>
using namespace std ;

#define SET(arr, what)  memset(arr, what, sizeof(arr))
#define FF(i, a)        for(i=0; i<a; i++)
#define SD(a)           scanf("%d", &a)
#define SSD(a, b)       scanf("%d%d", &a, &b)
#define SF(a)           scanf("%lf", &a)
#define SS(a)           scanf("%s", a)
#define SLD(a)          scanf("%lld", &a)
#define PF(a)           printf("%d\n", a)
#define PPF(a, b)       printf("%d %d\n", a, b)
#define SZ(arr)         (int)a.size()
#define SWAP(a,b)       a=a xor b;b= a xor b;a=a xor b;
#define read            freopen("in.txt", "r", stdin)
#define write            freopen("out.txt", "w", stdout)
#define MAX             1<<30
#define ESP             1e-5
#define lson            l, m, rt<<1
#define rson            m+1, r, rt<<1|1
template<class T> inline T sqr(T a){return a*a;}
template<class T> inline void AMin(T &a,T b){if(a==-1||a>b)a=b;}
template<class T> inline void AMax(T &a,T b){if(a<b)a=b;}
template<class T> inline T Min(T a,T b){return a>b?b:a;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
const int maxn = 55555 ;
int loy[maxn<<2], hash[1000001], head[maxn], k, ans[maxn] ;
int l[maxn], r[maxn] ;
struct node{
    int loy, blt, sup ;
}q[maxn] ;
struct edge{
    int v, nex ;
}e[maxn] ;
int cmp(const void *a, const void *b){
    return (*(node *)a).blt > (*(node *)b).blt ? -1 : 1 ;
}
void addedge(int a,int b){
    e[k].v = b ;
    e[k].nex = head[a] ;
    head[a] = k ;
    k ++ ;
}
void dfs(int x){
    l[x] = k ++ ;
    for(int i=head[x]; i; i=e[i].nex)
        dfs(e[i].v) ;
    r[x] = k ;
}
int query(int L, int R, int l, int r, int rt){
    if(r<l) return -1 ;
    if(L<=l&&r<=R){
        return loy[rt] ;
    }
    int m = (l + r) >> 1 ;
    int ret = -1 ;
    if(L<=m)    ret = Max(ret, query(L, R, lson)) ;
    if(m<R)     ret = Max(ret, query(L, R, rson)) ;
    return ret ;
}
void update(int idx, int val, int l, int r, int rt){
    if(l==r){
        loy[rt] = val ;
        return ;
    }
    int m = (l + r) >> 1 ;
    if(idx<=m)  update(idx, val, lson) ;
    else        update(idx, val, rson) ;
    loy[rt] = Max(loy[rt<<1], loy[rt<<1|1]) ;
}
int main(){
    int t, n, m, i, j ;
    SD(t) ;
    while(t--){
        SET(head, 0) ;
        SSD(n, m) ;
        k = 1 ;
        for(i=1; i<n; i++){
            SD(q[i].sup) ;SSD(q[i].loy, q[i].blt) ;
            hash[q[i].loy] = i ;
            addedge(q[i].sup, i) ;
        }
        k = 1 ;
        dfs(0) ;
        SET(loy, -1) ;
        qsort(q+1, n-1, sizeof(node), cmp) ;
        for(i=1; i<n; i=j){
            j = i ;
            while(j<n&&q[i].blt==q[j].blt){
                int id = hash[q[j].loy] ;
                int temp = query(l[id]+1, r[id]-1, 1, k-1, 1) ;
                if(temp==-1)    ans[id] = -1 ;
                else    ans[id] = hash[temp] ;
                j ++ ;
            }
            j = i ;
            while(j<n&&q[i].blt==q[j].blt){
                int id = hash[q[j].loy] ;
                update(l[id], q[j].loy, 1, k-1, 1) ;
                j ++ ;
            }
        }
        FF(i, m){
            SD(j) ;
            PF(ans[j]) ;
        }
    }
    return 0 ;
}