现在的位置: 首页 > 综合 > 正文

[各种面试题] 谷歌上机笔试–分数

2018年04月12日 ⁄ 综合 ⁄ 共 1281字 ⁄ 字号 评论关闭
题目描述:

有一棵无限完全二叉树,他的根节点是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

刚开始一直在想有什么规律,结果想半天看不出有什么规律,最后看别人提示,突破口在与隐含的完全二叉树这个条件。。。

对于给定的N或在 p/q,都能找到它到根的这么一条路径,有了这条路径就好做了。

所以复杂度是O(logN)的。

#include<iostream>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<climits>
#include<algorithm>
using namespace std;
void calNth(int n,int& p,int& q)
{
	vector<int> path;
	while(n!=1)
	{
		path.push_back(n%2);
		n>>=1;
	}
	p=1,q=1;
	for(int i=path.size()-1;i>=0;i--)
	{
		int t=p+q;
		if(path[i])
			p=t;
		else
			q=t;
	}
}
int solve(int p,int q)
{
	vector<int> path;
	while(p!=1||q!=1)
	{
		if(p>q)
		{
			path.push_back(1);
			p=p-q;
		}
		else
		{
			path.push_back(0);
			q=q-p;
		}
	}
	int cnt=0;
	int k=1,n=1;
	for(int i=path.size()-1;i>=0;i--)
	{
		cnt+=n;
		n<<=1;
		k=(k-1)*2+1+path[i];
	}
	cnt+=k;
	return cnt;
}

int main()
{
	int t;
	while(scanf("%d",&t)!=EOF)
	{
		int type=0;
		scanf("%d",&type);
		if(type==1)
		{
			int n;
			scanf("%d",&n);
			int p,q;
			calNth(n,p,q);
			printf("%d %d\n",p,q);
		}
		else
		{
			int p,q;
			scanf("%d%d",&p,&q);
			int n=solve(p,q);
			printf("%d\n",n);
		}
	}
}

抱歉!评论已关闭.