九度online judge 1543 二叉树

题目1543：无限完全二叉树的层次遍历

时间限制：1 秒

内存限制：128 兆

特殊判题：否

提交：389

解决：54

题目描述：

有一棵无限完全二叉树，他的根节点是1/1,且任意一个节点p/q的左儿子节点和右儿子节点分别是，p/(p+q)和(p+q)/q。如下图：

它的层次遍历结果如下：
1/1, 1/2, 2/1, 1/3, 3/2, 2/3, 3/1,...
有如下两类问题：
1.找到层次遍历的第n个数字。如，n为2时，该数字为1/2;
2.给定一个数字p/q,输出它在层次遍历中的顺序，如p/q为1/2时，其顺序为2;

输入：

输入包含多组测试用例，输入的第一行为一个整数T，代表共有的测试用例数。
接下去T行每行代表一个测试用例，每个测试用例有如下两种类型
1.1 n。输出层次遍历中，第n个数字。
2.2 p q。输出p/q在层次遍历中的顺序。
1 ≤ n, p, q ≤ 2^64-1

输出：

对于每个测试用例，若其类型为1，输出两个整数p q，代表层次遍历中第n个数字为p/q。
若其类型为2，输出一个整数n，代表整数p/q在层次遍历的中的顺序n。
数据保证输出在[1,2^64-1]范围内。

样例输入：

4
1 2
2 1 2
1 5
2 3 2

样例输出：

1 2
2
3 2
5

本来以为这道题是找规律的，还是自己对二叉树的概念不熟悉，这道题涉及到usigned long long

#include<stdio.h>
#include<iostream>
#include<cstring>
using namespace std;
int visit[1000];
int main()
{
	int T,i,choice,num;
unsigned long long sum,p,q,n;
	cin>>T;
	while(T--)
	{	num=0;
		sum=1;
		memset(visit,-1,sizeof(visit));
		cin>>choice;
		if(choice==1)
		{   p=q=1;
			scanf("%llu",&n);
			while(n!=1)
			{
				if(n%2==0)
				{
					n=n/2;
					visit[num++]=0;
				}
				else
				{
					n=(n-1)/2;
					visit[num++]=1;
				}
			}
			for(i=num-1;i>=0;i--)
			{
				if(visit[i]==0)
				{
					q=p+q;
					p=p;
				}
				if(visit[i]==1)
				{
					q=q;
					p=p+q;
				}
			}
			printf("%llu %llu\n",p,q);
		}
		if(choice==2)
		{
			scanf("%llu%llu",&p,&q);

			while(p-q!=0)
			{
				if(p>q)
				{
					visit[num++]=1;
					p=p-q;
					q=q;
				}
				if(p<q)
				{
					visit[num++]=0;
					p=p;
					q=q-p;
				}
			}
			for(i=num-1;i>=0;i--)
			{
				if(visit[i]==0)
				{
					sum=sum*2;
				}
				if(visit[i]==1)
				{
					sum=sum*2+1;
				}
			}
			printf("%llu\n",sum);

		}
	}
	return 0;
}