HDU 4605 Magic Ball Game (dfs+离线树状数组)

 题意:给你一颗有根树,它的孩子要么只有两个,要么没有,且每个点都有一个权值w。

接着给你一个权值为x的球,它从更节点开始向下掉,有三种情况

x=w[now]:停在此点

x<w[now]:当有孩子时:1/2可能性到左孩子,1/2可能性到右孩子

x>w[now]:当有孩子时:1/8可能性到左孩子,7/8可能性到右孩子

再给你一个点U,问你从根节点到U结点的可能性为多少 7^x/2^y ,求出x y

题解:非常经典的一个题,我使用邻接表存储,接着使用dfs遍历,最后树状数组维护结果。

我们可以知道    1/2:y++        1/8:y+=3     7/8:x++,y+=3

接着就是每次只需要找到唯一一条从根到U点父节点的所有权值

接着找:左孩子中小于X的个数,右孩子中小于X的个数,左孩子中大于X的个数,右孩子中大于X的个数,但是我们要注意一个情况就是当有等于X的时候就不能到达U

这样我们可以离线vector存储问题(同一个点不同则存在一起),从根节点开始dfs遍历每个点

当走孩子节点时就把此点的权值加入数组(开两个数组),回溯时删除此权值

在添加前就此点权值寻找上诉四个值,并利用树状数组维护

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define eps 1E-8
/*注意可能会有输出-0.000*/
#define Sgn(x) (x<-eps? -1 :x<eps? 0:1)//x为两个浮点数差的比较,注意返回整型
#define Cvs(x) (x > 0.0 ? x+eps : x-eps)//浮点数转化
#define zero(x) (((x)>0?(x):-(x))<eps)//判断是否等于0
#define mul(a,b) (a<<b)
#define dir(a,b) (a>>b)
typedef long long ll;
typedef unsigned long long ull;
const int Inf=1<<28;
const double Pi=acos(-1.0);
const int Mod=1e9+7;
const int Max=200010;//包括询问
int head[Max],nnext[Max],to[Max],chi[Max],e;//邻接表存树 dfs遍历
struct node
{
    int val,poi;
} que[Max]; //问题
int dir[3][2]= {{0,1},{0,3},{1,3}}; //比x大的孩子 比x小的左孩子 比x小的右孩子
map<int,int> mp;//离散化
int val[Max];
vector<int> vec[Max];//存下这个点有哪些权值
int len[Max],ansx[Max],ansy[Max];
int nn;//总权值点
void Init(int n,int *lbit,int *rbit)
{
    for(int i=0; i<=n; ++i)
    {
        len[i]=0;
        ansx[i]=0;
        ansy[i]=0;
        head[i]=-1;
        vec[i].clear();
    }
    e=0;
    nn=0;
    mp.clear();
    memset(lbit,0,sizeof(lbit));
    memset(rbit,0,sizeof(rbit));
    return;
}
void AddEdge(int u,int v,int ch)
{
    nnext[e]=head[u];
    to[e]=v;
    chi[e]=ch;
    head[u]=e++;
    return;
}
int lowbit(int x)
{
    return x&(-x);
}
void Add(int x,int y,int *bit)
{
    while(x<=nn)
    {
        bit[x]+=y;
        x+=lowbit(x);
    }
    return;
}
int Sum(int x,int *bit)
{
    int sum=0;
    while(x)
    {
        sum+=bit[x];
        x-=lowbit(x);
    }
    return sum;
}
int bit[2][Max];//区分左右孩子
void dfs(int son,int tol1,int tol2)//遍历邻接表
{
    int now=mp[val[son]];
    for(int i=0; i<len[son]; ++i)
    {
        int now1=mp[que[vec[son][i]].val];
        int lef1=Sum(now1,bit[0]);
        int lef2=Sum(now1-1,bit[0]);
        int rig1=Sum(now1,bit[1]);
        int rig2=Sum(now1-1,bit[1]);
        if(lef1-lef2||rig1-rig2)
        {
            ansx[vec[son][i]]=-1;
        }
        else//求值
        {
            ansx[vec[son][i]]=rig2*dir[2][0];

            ansy[vec[son][i]]=(tol1-lef1)*dir[0][1];
            ansy[vec[son][i]]+=(tol2-rig1)*dir[0][1];
            ansy[vec[son][i]]+=lef2*dir[1][1];
            ansy[vec[son][i]]+=rig2*dir[2][1];
        }
    }
    for(int i=head[son]; ~i; i=nnext[i])
    {
        Add(now,1,bit[chi[i]]);//加点
        if(!chi[i])
        dfs(to[i],tol1+1,tol2);
        else
            dfs(to[i],tol1,tol2+1);
        Add(now,-1,bit[chi[i]]);//删点
    }
    return;
}
int main()
{
    int t,n,m,q;
    int u,a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        Init(n,bit[0],bit[1]);
        for(int i=1; i<=n; ++i)
        {
            scanf("%d",&val[i]);
            mp[val[i]]=1;
        }
        scanf("%d",&m);
        for(int i=0; i<m; ++i)//邻接表
        {
            scanf("%d %d %d",&u,&a,&b);
            AddEdge(u,a,0);
            AddEdge(u,b,1);
        }
        scanf("%d",&q);
        for(int i=0; i<q; ++i)
        {
            scanf("%d %d",&que[i].poi,&que[i].val);
            vec[que[i].poi].push_back(i);//根据树上某点存que小标
            len[que[i].poi]++;
            mp[que[i].val]=1;
        }
        int cnt=1;
        for(map<int,int>::iterator it=mp.begin(); it!=mp.end(); ++it) //离散化
        {
            it->second=cnt++;
            nn++;
        }
        dfs(1,0,0);
        for(int i=0; i<q; ++i)
        {
            if(ansx[i]==-1)
                printf("0
");
            else
                printf("%d %d
",ansx[i],ansy[i]);
        }
    }
    return 0;
}
原文地址:https://www.cnblogs.com/zhuanzhuruyi/p/5910396.html