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.