E: A Rational Sequence 2
题意理解:就是让你在满二叉树上寻找规律,根据给出的p,q推出对应的编号;
利用每个左子节点编号是父节点遍号 * 2 ,右子节点编号是父节点编号 * 2 + 1
- 1 并且如果 p > q那么是往上找到第一个p < q的节点一直沿其右节点走
那么这个节点的编号是往上找到第一个p < q的节点的编号 * pow(2,k) + pow(2,k) - 1,k是起始节点向下多少层到目标节点(毕竟每向下一层都要*2 + 1,最后就能推出来上面那个规律) - 2 如果是p < q那么就向上找第一个 p > q的节点,那么是往上找到第一个p > q的节点一直沿其左节点走那么编号就是其编号 * pow(2,k)
- 3注意特判是不是会找到根节点
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <deque>
#include <ext/rope>
#include <iostream>
#include <queue>
#include <stack>
#include <string>
#include <cmath>
#include <vector>
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int maxn = 1e6 + 5;
const int INF = 1e9 + 7;
ll DFS(int p,int q)
{
ll k,a,ans;
if(!p||!q) return 1;
if(p == 1&&q== 1) return 1;
//右节点
if(p > q){
k = p / q;//看它到以第一个p < q的节点为根时要向下多少层
if(p % q == 0) k--;//一路到根时,要特判
a = DFS(p % q,q);//求第一个p < q的节点编号
//以编号为a的根节点一直向右下k层的的右子节点编号
ans = pow(2 , k) * a + pow(2,k) - 1;//每个节点的右子节点编号是自己编号 * 2 + 1
}
//这个节点为向上遇到第一个p >q 的节点向左下走的节点或者为根一直向左下的节点
if(p < q){
k = q / p;
if(q % p == 0) k--;
a = DFS(p , q % p);
//每个节点的左子节点编号是自己编号 * 2 ,所以是2的k次方因为它是一直沿着左子节点走
ans = pow(2 ,k) * a;
}
return ans;
}
int main()
{
int t;
cin>>t;
while(t--){
int k,p,q;
scanf("%d%d/%d",&k,&p,&q);
printf("%d %lld\n",k,DFS(p,q));
}
return 0;
}