B. Special Numbers

B. Special Numbers

题目链接

题意
给你一个底数n，让你算出排在第k个以n为底数形式不同的幂的和

解题思路
手动模拟一下你会发现，第一个数是$1$，第二个数是$n+1$，第三个数是$n^2$,第四个数是$n^2+1$，那么你会发现，它其实就是二叉树的排列，给的k就是十进制下，在二叉树中的第k个结点，根据这个我们构造出该结点的所有有关的结点，回溯到二叉树构造的问题，从根结点开始，与当前它的父节点比较，相等就是乘以$n$,不相等丢给右孩子乘以$n$再加一。

ACcode

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me(a) memset(a,0,sizeof(a)) 
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pi pair<ll,ll>
using namespace std;
const int N = 1e5+10;
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;} 
const int mod = 1e9+7;
int main() {
     int t;
     std::cin>>t;
     while(t--){
        int n,k;
        std::cin>>n>>k;
        std::vector<int> v;
        while(k){
          v.pb(k);
          k/=2;
        }
        reverse(v.begin(), v.end());
        ll res=1;
        int cur=1;
        for(int i=1;i<(int)v.size();i++){
            if(v[i]==cur*2){
              res=(res*n%mod)%mod;
              cur*=2;}
            else {
              res=(res*n%mod+1)%mod;
              cur=cur*2+1;}
        }
         std::cout<<res<<'\n';
     }
    return 0;
  }