Weak Pair HDU - 5877

点击打开链接

dfs遍历到每一个点 u 时 dfs序列中存的就是这个点的所有祖先 v 了 我们只需要知道有多少祖先的值 per[v] 小于 k/per[u] 而这个功能用线段树实现即可

还有就是需要离散化 先将所有 per[i ]与 k/per[i] 存入数组 然后排序去重 以此数组的值建立线段树(也可通过map建立映射关系) 详见代码

#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 0x3f3f3f3f3f3f3f3f

struct nodeI
{
int v;
int next;
};

struct nodeII
{
ll l;
ll r;
int val;
int flag;
};

nodeI edge[100010];
nodeII tree[800010];
ll per[100010],pre[200010];
int first[100010],book[100010];
ll k,ans;
int n,m,num;

void addedge(int u,int v)
{
edge[num].v=v;
edge[num].next=first[u];
first[u]=num++;
return;
}

void pushup(int cur)
{
tree[cur].val=tree[cur*2].val+tree[cur*2+1].val;
return;
}

void build(int l,int r,int cur)
{
int m;
tree[cur].l=pre[l];
tree[cur].r=pre[r];
tree[cur].val=0;
tree[cur].flag=0;
if(l==r)
{
tree[cur].flag=1;
return;
}
m=(l+r)/2;
build(l,m,cur*2);
build(m+1,r,cur*2+1);
return;
}

void update(ll tar,int val,int cur)
{
if(tree[cur].flag)
{
tree[cur].val+=val;
return;
}
if(tar<=tree[cur*2].r) update(tar,val,cur*2);
else update(tar,val,cur*2+1);
pushup(cur);
return;
}

ll query(ll pl,ll pr,int cur)
{
ll res;
if(pl<=tree[cur].l&&tree[cur].r<=pr)
{
return tree[cur].val;
}
res=0;
if(pl<=tree[cur*2].r) res+=query(pl,pr,cur*2);
if(pr>=tree[cur*2+1].l) res+=query(pl,pr,cur*2+1);
return res;
}

void dfs(int u)
{
ll p;
int i,v,a,b;
if(per[u]==0) p=N;
else p=k/per[u];
ans+=query(0,k/per[u],1);
update(per[u],1,1);
for(i=first[u];i!=-1;i=edge[i].next)
{
v=edge[i].v;
if(!book[v])
{
book[v]=1;
dfs(v);
}
}
update(per[u],-1,1);
return;
}

int main()
{
int t,i,j,u,v;
scanf("%d",&t);
while(t--)
{
scanf("%d%lld",&n,&k);
for(i=1;i<=n;i++)
{
scanf("%lld",&per[i]);
pre[i]=per[i];
}
for(i=n+1;i<=2*n;i++)
{
if(per[i-n]==0) pre[i]=N;
else pre[i]=k/pre[i-n];
}
sort(pre+1,pre+2*n+1);
pre[0]=-1;
for(i=1,j=0;i<=2*n;i++)
{
if(pre[j]!=pre[i])
{
pre[++j]=pre[i];
}
}
m=j;
build(1,m,1);

memset(first,-1,sizeof(first));
memset(book,0,sizeof(book));
num=0;
for(i=1;i<=n-1;i++)
{
scanf("%d%d",&u,&v);
addedge(u,v);
book[v]=1;
}
ans=0;
for(i=1;i<=n;i++)
{
if(!book[i])
{
memset(book,0,sizeof(book));
book[i]=1;
dfs(i);
break;
}
}
printf("%lld\n",ans);
}
return 0;
}