HDU 5877 Weak Pair

题意；求所有点的祖先节点中与他的权值的乘积小于k 的个数的和；

思路：我们在从祖先节点开始进行DFS， 当跑到该节点时，在线段树中查找比k/val[u] 小的权值的个数，并在线段树中该权值对应的点+1, 当回溯的时候该点权值的位置在-1。

由于权值的范围较大，可以先将权值离散化，或者直接建立一颗手动分配内存的线段树，如果离散化树状数组也可解，当然维护一个set也是可以的。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int nn = 100100;
const int maxx = 1000000007;

struct node
{
int l, r, ls, rs, v;
};
struct EDGE
{
int to, next;
};
node t[nn * 100];
EDGE g[nn * 3];
int n, sz, siz, root;
ll k;
int head[nn], val[nn];
void add_edge(int u, int v)
{
g[sz].to = v;
g[sz].next = head[u];
head[u] = sz++;
}

int now_t(int l,int r)
{
t[siz].l = l;
t[siz].r = r;
t[siz].ls = -1;
t[siz].rs = -1;
t[siz++].v = 0;
return siz - 1;
}

void pushup(int rt)
{
t[rt].v = 0;
if (t[rt].ls != -1) t[rt].v += t[t[rt].ls].v;
if (t[rt].rs != -1) t[rt].v += t[t[rt].rs].v;
}
void insert(int x, int rt,int val)
{
if (t[rt].l == t[rt].r)
{
t[rt].v += val;
return;
}
int mid = (t[rt].l + t[rt].r) >> 1;
if (x <= mid)
{
if (t[rt].ls == -1) t[rt].ls = now_t(t[rt].l, mid);
insert(x, t[rt].ls, val);
}
else
{
if (t[rt].rs == -1) t[rt].rs = now_t(mid + 1, t[rt].r);
insert(x, t[rt].rs, val);
}
pushup(rt);
}
int find(int l,int r, int rt)
{
if (l <= t[rt].l && t[rt].r <= r)
return t[rt].v;
int mid = (t[rt].l + t[rt].r) >> 1;
int ans = 0;
if (mid >= l && t[rt].ls !=-1)
ans += find(l, r, t[rt].ls);
if (mid < r && t[rt].rs != -1)
ans += find(l, r, t[rt].rs);
return ans;
}

ll cnt;
void dfs(int u, int fa)
{
ll r = k / val[u];
if (r > maxx) r = maxx-1;
cnt +=(ll) find(1, r, root);
insert(val[u], root, 1); //***************
for (int i = head[u]; i != -1; i = g[i].next)
{
int v = g[i].to;
if (v == fa) continue;
dfs(v, u);
}
insert(val[u], root, -1);
}
int pos[nn];
int main()
{
int tt;
scanf("%d", &tt);
while (tt--)
{
scanf("%d%lld", &n, &k);
for (int i = 1; i <= n; i++)
scanf("%d", &val[i]);
memset(head,-1,sizeof(head));
memset(pos,0,sizeof(pos));
sz = 0;
siz = 1;
cnt = 0;
root = now_t(1, maxx);
for (int i = 1; i < n; i++)
{
int u, v;
scanf("%d%d", &u, &v);
add_edge(u, v);
add_edge(v, u);
pos[v]++;
}
int st;
for(int i=1;i<=n;i++)
{
if(pos[i]==0)
{
st=i;
break;
}
}
dfs(st, -1);
printf("%lld\n",cnt);
}
return 0;
}

这题有个小坑就是，树的根节点不一定为1.