这一题开始一看到那么大的数就懵逼了,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));
}
}