代码题--C++--求数列的第n项牛客网第n个数所在的那一层之前的所有层里共有多少个数
题目描述:
米兔从兔米那里了解到有一个无限长的数字序列 1, 2,3,3,4,4,4, 5,5,5,5,5 ...,(已知此数列有一定规律,现将这些数字按不同数值堆叠,相同值的数字在同一层)。米兔想知道这个数字序列的第n个数所在的那一层之前的所有层里共有多少个数。
输入描述:
n(n<=1e18)
输出描述:
第n个数所在的那一层之前的所有层里共有多少个数
示例1
输入
6
输出
4
解题思路:
肯定是先找规律,如下:因为输入的n是第几个数字,这就要算和,累计欠两个数的之和就是a+b,然后和n比较,然后用n>=a+b先找到n的位置,因为要求那一层之前的所有层里共有多少个数,所以在找的过程中存储前面数字个数的和,也就是sum = sum+c,c = a+ b;然后用a,b,c传递,一行一行移动。具体可以调试的时候看出算法,在while的条件时,我测试了,如果不写后面的n<b,也行,因为比如9,因为 9 >=(1+1+2+3)+5是错误的,所有要执行9<5,它也是错误的,所以n<b,就是要代表,n也满足小于b的这个条件。
a从0开始,一直想下一行移动,保存的是当前行有多少个数,a继承b行中数字的个数,也就是下图的蓝色数字
b从1开始,一直想下一行移动,保存的是当前行有多少个数,b继承c行中数字的个数,也就是下图的蓝色数字
c保存下一行的数字个数,也就是前一行和前前一行的和,即a+b;
sum保存的是累计到包括此数字从在的行数之前的所有数字之和, n = 9,sum= 1+1+2+3+5=12
具体代码:
#include <iostream>
using namespace std;
int main()
{
long long int n ;
cin >> n;
long long int a = 0,b = 1, c = 1;
long long int sum = 0;
while(n >= a+b || n < b)
{
sum = sum + c;
c = a + b;
a = b;
b = c;
}
cout << sum - a <<endl;
return 0;
}