HDU - 5877 Weak Pair (dfs+树状数组)

题目链接：Weak Pair

题意：

　　给出一颗有根树，如果有一对u,v，如果满足u是v的父节点且vec[u]×vec[v]<=k，则称这对结点是虚弱的，问这棵树中有几对虚弱的结点。

题解：

　　刚开始看到这题，无脑暴力dfs从叶子结点向上递归，TLE了一发神清气爽@。@！所以用树状数组优化dfs，从根节点向下递归，在结点x找所有满足<=k/vec[x]的点的个数。这里因为数据很大要离散化一下，在输入的时候就把vec[x]和M/vec[x]都放数组里面，sort一下后再放到map里面。这题学了一下用树状数组优化dfs，感觉挺好的～

 #include<bits/stdc++.h>
 using namespace std;
 const int MAX_N = 1e5+;
 vector<int> P[MAX_N];
 int in[MAX_N];
 long long vec[MAX_N*];
 long long vec1[MAX_N*];
 long long res[MAX_N*];
 long long pos[MAX_N*];
 map<long long ,int > mp;
 long long ans = ;
 long long N,M,T;
 void add(int x,long long num)
 {
     for(;x<*MAX_N;x+=(x&-x))
         res[x] += num;
 }
 long long sum(int x)
 {
     long long ans = ;
     for(;x>;x-=(x&-x))
         ans += res[x];
     return ans;
 }
 void dfs(int x)
 {
     ans += sum(mp[M/vec[x]]);
     add(mp[vec[x]],);
     for(int i=;i<P[x].size();i++)
     {
         dfs(P[x][i]);
     }
     add(mp[vec[x]],-);
 }
 int main()
 {
     cin>>T;
     while(T--)
     {
         mp.clear();
         memset(in,,sizeof(in));
         memset(res,,sizeof(res));
         for(int i=;i<MAX_N;i++) P[i].clear();
         cin>>N>>M;
         for(int i=;i<=N;i++)
         {
             scanf("%lld",&vec[i]);
             vec1[i+N] = M/vec[i];
             vec1[i] = vec[i];
         }
         sort(vec1+,vec1+*N+);
         for(int i=;i<=*N;i++)
         {
             mp.insert(make_pair(vec1[i],i));
         }
         for(int i=;i<N-;i++)
         {
             int a,b;
             scanf("%d%d",&a,&b);
             P[a].push_back(b);
             in[b] ++;
         }
         int point = ;
         for(int i=;i<=N;i++)
         {
             if(!in[i])
             {
                 point = i;
                 break;
             }
         }
         ans = ;
         dfs(point);
         cout<<ans<<endl;
     }
     return ;
 }