题目描述
在一棵树上有n只鸟。由于感受到了自然的力量,它们希望学习自然数。因此它们站在树上,演唱所有自然数。自然数是从1开始的,按照递增的顺序演唱。当演唱到K时,会有K只鸟受不了而飞走。如果演唱到某个自然数而鸟数不足时,它们会从1开始重新演唱。假设每秒钟它们都会演唱一个自然数,请你求一求,所有鸟都飞走需要多长时间。
【样例说明】
第一秒,一只鸟飞走。第二秒,两只鸟飞走。第三秒时,需要三只鸟,但只剩一只鸟,因此它会从1开始演唱,并且飞走。
输入格式
一个整数n(1<=n<=10^9),表示鸟数。
输出格式
一个整数t,表示总时间。
rqnoj十月苦情赛。被rqnoj坑了,本来220的,被吃了100分。
加注释处是最早版本,后来才发现那个公式可以化简。
#include <cmath>
#include <iostream>
using std::cout;
using std::cin;
#include <cstdio>
const long oo = 0x7fff0000;
long long n;
long long tot = 0;
int main()
{
cin >> n;
while (n>0)
{
long t = (long((-1ll+sqrt(1ll+(n<<3ll))))>>1ll);
tot += t;
// n -= (((t<<1)+1)*((t<<1)+1)-1)>>3;
n -= ((t*(t+1))>>1);
}
cout << tot;
return 0;
}