Constraints
Time Limit: 1 secs, Memory Limit: 32 MB
Description
给定一个正整数k(3≤k≤15),把所有k的方幂及所有有限个互不相等的k的方幂之和构成一个递增的序列,例如,当k=3时,这个序列是:
1,3,4,9,10,12,13,…
(该序列实际上就是:30,31,30+31,32,30+32,31+32,30+31+32,…)
请你求出这个序列的第N项的值(用10进制数表示)。
例如,对于k=3,N=100,正确答案应该是981。
Input
输入包含多个测试数据。
每个测试数据只有1行,为2个正整数,用一个空格隔开:
k N
(k、N的含义与上述的问题描述一致,且3≤k≤15,10≤N≤1000)
Output
对于每个测试数据输出一个正整数(在所有的测试数据中,结果均不超过2.1*109)。
Sample Input
3 100
3 100
Sample Output
981
981
解题思路:
首先考虑一个方幂数列,每增加一个数字,则可以利用该数字加上若干个该方幂数列前面的数字组成一个新的数字,考虑用开辟一个数组保留结果数列,则利用动态规划的思想不难得到:所增加的数字正好是方幂数列新增的数字加上结果数列之前所有数字。
#include <cstdio>
#include <vector>
using namespace std;
int main(){
int k, n;
while(scanf("%d%d", &k, &n) != EOF){
int prePower = 1;
vector<int> v;
v.reserve(1009);
v.push_back(1);
int cnt = 1;
while(cnt <= n){
prePower *= k;
v.push_back(prePower);
cnt++;
for(int i = 0, end = v.size()-1; i < end && cnt <= n; i++){
v.push_back(prePower + v[i]);
cnt++;
}
}
printf("%d\n", v[n-1]);
}
return 0;
}
此外,有另外一种很巧妙的做法,以k=3为例:
3^0, 3^1, 3^2, ……
题目可以看做是对这个数列求个有序的排列,即
若n=1,则取1(二进制)第一个数;
若n=2,则取10(二进制),第二个数;
若n=3,则取11(二进制),第一个数和第二个数的和;
…….
因此只需要求出n的二进制形式,然后在有1的位上加上以k为权值的数即可
#include <cstdio>
using namespace std;
int main(){
int k, n;
while(scanf("%d%d", &k, &n) != EOF){
int p = 1, res = 0;
while(n > 0){
if(n&1){
res += p;
}
p *= k;
n = n >> 1;
}
printf("%d\n", res);
}
return 0;
}