AcWing 428. 数列(数学/二进制)

题目链接

给定一个正整数k,把所有k的方幂及所有有限个互不相等的k的方幂之和构成一个递增的序列，例如，当k=3时，这个序列是：

1，3，4，9，10，12，13，…

该序列实际上就是： $3^0$ ，$3^1$，$3^0$+$3^1$，$3^2$，$3^0$+$3^2$，$3^1$+$3^2$，$3^0$+$3^1$+$3^2$，…
请你求出这个序列的第N项的值（用10进制数表示）。

例如，对于k=3，N=100，正确答案应该是981。

输入格式

输入文件只有1行，为2个正整数，用一个空格隔开：k N。

输出格式

输出文件为计算结果，是一个正整数（在所有的测试数据中，结果均不超过2.1∗109）。（整数前不要有空格和其他符号）。

数据范围

3≤k≤15,
10≤N≤1000

输入样例：

3 100

输出样例：

981

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int N = 1e5 + 10;
const int M = 111;
using namespace std;

ll get_res(int x,int y){
    ll res=1;
    while(y--) res*=x;
    return res;
}

void solve(){
    int n,k;
    cin>>k>>n;
    ll res=0;
    rep(i,0,19){
        if(n>>i&1) res+=get_res(k,i);
    }
    cout<<res<<endl;
}

int main() {
    solve();
	return 0;
}