[NOIP2006]数列
链接:https://ac.nowcoder.com/acm/contest/19305/1019
来源:牛客网
题目描述
给定一个正整数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。
输入描述:
输入1行,为2个正整数,用一个空格隔开:k N(k、N的含义与上述的问题描述一致,且3 ≤ k ≤ 15,10 ≤ N ≤ 1000 )。
输出描述:
输出一个正整数(在所有的测试数据中,结果均不超过2.1*109)。(整数前不要有空格和其他符号)。
示例1
输入
复制
3 100
输出
复制
981
总结:采用深度搜索,其中还要注意时间的限制,比如我习惯性的用i++,结果半天结果不出来,然后改成i+1,结果一下子就出来了。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string>
#include<string.h>
#include<algorithm>
#include<vector>
#include<string>
#include<map>
#include<set>
#include<cmath>
#include<math.h>
#include<queue>
#include<stack>
using namespace std;
typedef long long LL;//数据比较大采用long long.
int n,k;
set<LL>p;
void solve(LL sum,int step)
{
if(sum>2.1e10)//结果限制
return;
p.insert(sum);
if(step<15){
solve(sum+pow(k,step),step+1);
solve(sum,step+1);
}
}
int main()
{
cin>>k>>n;
solve(0,1);//初始化结果为0
solve(1,1);//初始化结果为1
int i=0;
for(auto t=p.begin();t!=p.end();t++)//set遍历
{
if(i==n)
cout<<*t<<endl;
i++;
}
return 0;
}
看到有个大佬是这样子写的,感觉非常巧妙,膜拜大佬。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string>
#include<string.h>
#include<algorithm>
#include<vector>
#include<string>
#include<map>
#include<set>
#include<cmath>
#include<math.h>
#include<queue>
#include<stack>
using namespace std;
typedef long long LL;
LL n,k;
int main()
{
cin>>k>>n;
LL sum=0;
for(int i=0;i<=30;i++)
{
if(n>>i&1)
sum+=pow(k,i);
}
cout<<sum<<endl;
return 0;
}
算法实在是太巧妙了,希望能学到更多的东西。