4605 Magic Ball Game

先离线记录所有询问，然后用dfs在线记录当前节点的所有父亲结点，父亲结点分为两类，一个从左边过来，一个从右边过来，分别开一个数组。

数据需要离散化，刚开始用lower_bound写，一直wa。。，用二分就过了。

之后总结了下lower_bound和upper_bound的规律。

#include<stdio.h>#include<algorithm>#include<string.h>using namespace std;int a[7]={5,9,13,19,25,39,100};int main(){int i,j,k,n;while(scanf("%d",&n))    {  printf("%d\n",lower_bound(a,a+7,n)-a);//第一个大于等于他的数的下标，当不存在大于等于他的数时，返回数组最后一个数下标加一      printf("%d\n",upper_bound(a,a+7,n)-a);//第一个大于他的数的下标，当不存在大于他的数时，返回数组最后一个数下标加一    }               }

下面是代码

#pragma comment(linker, "/STACK:1024000000,1024000000")#include<stdio.h>#include<string.h>#include<algorithm>#include<iostream>#include<stdlib.h>#include<vector>#define LL long longusing namespace std;const int maxn=100005;int cl[100005],cr[100005],p[100005],quar,len;int w[maxn],numl=0,numr=0,flag;vector<int>son[100005];vector< pair<int,int> >q[100005];struct Q{    int x,y;}qq[100005];int lowbit(int cur){    return cur&(-cur);}void update(int cur,int dir,int s){    if(dir==-1)      while(cur<maxn)      {        cl[cur]+=s;        cur+=lowbit(cur);      }    else        while(cur<maxn)       {           cr[cur]+=s;           cur+=lowbit(cur);       }}int getnum(int cur,int dir){    int ans=0;    if(dir==-1)    {        while(cur>0)        {            ans+=cl[cur];            cur-=lowbit(cur);        }    }    else    {        while(cur>0)        {            ans+=cr[cur];            cur-=lowbit(cur);        }    }    return ans;}int findcou(int cur){    int st=0,ed=len-1,mid;    if(p[ed]<cur)return len-1;    if(p[st]>cur)return -1;    while(st<ed)    {        mid=(st+ed)/2;        if(p[mid]==cur){flag=1;return mid;}        if(p[mid]>cur)ed=mid;        else   st=mid+1;    }    if(p[st]==cur)    {        flag=1;        return st;    }    return st-1;}void dfs(int s){    int i,j,k,l,r,x,y,cou,h;    for(i=0;i<q[s].size();i++)    {        k=q[s][i].first;       // printf("s=%d,k=%d,\n",s,k);        flag=0;        k=findcou(k)+1;        //printf("s=%d,k=%d,\n",s,k);        y=numl+numr;        l=getnum(k,-1);        r=getnum(k,1);        //printf("numl=%d,numr=%d,l=%d,r=%d\n",numl,numr,l,r);        y+=2*(l+r);x=r;        if(flag)        {        if(l>0)        {             l-=getnum(k-1,-1);             if(l>0)x=y=-1;        }        if(r>0)        {           r-=getnum(k-1,1);           if(r>0)x=y=-1;        }        }        cou=q[s][i].second;        //puts("ok??");        qq[cou].x=x;qq[cou].y=y;    }    //puts("ok??");    if(son[s].size()==0)return ;    int tmp=findcou(w[s])+1;    //printf("s=%d,tmp=%d\n",s,tmp);    update(tmp,-1,1);numl++;dfs(son[s][0]);    //printf("numl=%d,numr=%d,",numl,numr);    update(tmp,-1,-1);numl--;    update(tmp,1,1);numr++;dfs(son[s][1]);    update(tmp,1,-1);numr--;}int main(){    int j,m,n,v,x;    int t,i,tmp,u,a,b;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        for(i=1;i<=n;i++)        {           scanf("%d",&w[i]);           p[i-1]=w[i];           cl[i]=cr[i]=0;           son[i].clear();           q[i].clear();        }         sort(p,p+n);        len=unique(p,p+n)-p;        scanf("%d",&m);        for(i=0;i<m;i++)        {            scanf("%d%d%d",&u,&a,&b);            son[u].push_back(a);            son[u].push_back(b);           // printf("%d----%d\n",u,son[u].size());        }        scanf("%d",&quar);        for(i=0;i<quar;i++)        {            scanf("%d%d",&v,&x);            q[v].push_back(make_pair(x,i));            //printf("%d-----%d\n",i,q[v].size());        }        //for(i=0;i<=len;i++)            //printf("%d,",p[i]);        numr=numl=0;        dfs(1);        for(i=0;i<quar;i++)        if(qq[i].x==qq[i].y&&qq[i].x==-1)           puts("0");        else            printf("%d %d\n",qq[i].x,qq[i].y);    }    return 0;}