Problem E - A Rational Sequence 2015 ACM/ICPC Greater New York Region

时间:2021-11-10 20:38:56

这一题开始一看到那么大的数就懵逼了,BFS肯定会TLE。

根据子节点和父节点的关系可以从下往上递归找出p/q对应的层数。根据input中p,q的大小关系可以判定出是左节点还是右节点。因为是二叉树又是层序遍历,num of 第l层最右的node就是sum_{i=0}^l 2^l。如果p/q在l层,n=num of l-1层最右的node+(num of father-{l-1层第一个节点的num})*2+1(if 左节点) or 2(if 右节点)

用记忆化搜索即可,因为对应的二叉树是固定的,每个test case前不必清空,可以省时间。

PS:我开始以为p and q互质还敲了个quick gcd,后来发现并不要考虑这一点。

#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<string>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<cstring>
#include<vector>
#include<queue>
#include<map>

using namespace std;
int P;
int K;
int p;
int q;
map<pair<int,int>,int>level;
map<pair<int,int>,pair<int,int> >fa;
map<pair<int,int>,int>num;
int levelnum[50];
void init()
{
    memset(levelnum,0,sizeof(levelnum));
    int prod=1;
    levelnum[1]=1;
    for(int i=2;i<50;i++)
    {
        prod*=2;
        levelnum[i]=levelnum[i-1]+prod;

    }
}
int quickgcd(int a,int b)
{

    if(a==0) return b;
    if(b==0) return a;
    if(!(a&1)&&!(b&1))
    {

        return quickgcd(a>>1,b>>1)<<1;
    }
    else if(!(b&1))
    {
        return quickgcd(a,b>>1);
    }
    else if(!(a&1)) return quickgcd(a>>1,b);
    else return quickgcd(abs(a-b),min(a,b));
}
int dfslevel(int x,int y)
{
    //cout<<x<<" level "<<y<<endl;
    pair<int,int>p=make_pair(x,y);
    if(x==1&&y==1)
    {
        level[p]=1;
        fa[p]=make_pair(0,0);
        return level[p];
    }
    if(level[p]!=0)
    {
        return level[p];
    }
    else
    {
        if(x>y)//right
        { //use gcd later
//            int fax=x-y;
//            int fay=y;
//            int facfa=quickgsc(fax,fay);
//            fax=fax/facfa;
//            fay=fay/facfa;
            fa[p]=make_pair(x-y,y);
            level[p]=dfslevel(x-y,y)+1;
            return level[p];
        }
        else if(x<y)
        {
//            int fax=x-y;
//            int fay=y;
//            int facfa=quickgsc(fax,fay);
            fa[p]=make_pair(x,y-x);
            level[p]=dfslevel(x,y-x)+1;
            return level[p];
        }
    }
}
int dfsnum(int x,int y)
{
    pair<int,int>p=make_pair(x,y);
    if(x==1&&y==1)
    {
        return num[p]=1;
    }
    if(num[p]!=0)
    {
        return num[p];
    }
    if(x>y)//right
    {
        pair<int,int>q=fa[p];
        int falevel=level[q];
        int remain=(dfsnum(q.first,q.second)-levelnum[level[fa[q]]]-1)*2+2;
        num[p]=levelnum[falevel]+remain;
        return num[p];
    }
    if(x<y)//left
    {
        pair<int,int> q =fa[p];
        int falevel=level[q];
        int remain=(dfsnum(q.first,q.second)-levelnum[level[fa[q]]]-1)*2+1;
        num[p]=levelnum[falevel]+remain;
        return num[p];
    }

}
int main()
{
    //freopen("input.txt","r",stdin);
    //cout<<log(pow(2,32)-1)/log(2)<<endl;
    init();
    scanf("%d",&P);
    for(int ca=0;ca<P;ca++)
    {
        scanf("%d %d/%d",&K,&p,&q);
        int fac=quickgcd(p,q);
        p=p/fac;
        q=q/fac;//fac=1?
        //cout<<p<<" "<<q<<endl;
        int l=dfslevel(p,q);
        //cout<<l<<endl;
        printf("%d %d\n",K,dfsnum(p,q));
    }
}