练习题0006

题目描述

美国大选正在进行，zyk、川宝和拜登激烈的争夺选票。zyk的狂热粉丝们为了帮助zyk赢得胜利，制定了一套投票规则......

zyk的狂热粉丝们决定这样来投票，第一天，投一票；之后两天（第二天和第三天），每天投两票；之后三天（第四、五、六天），每天投三票；之后四天（第七、八、九、十天），每天投四票……；这个投票会一直这样延续下去：当连续N天每天投N张票后，粉丝们会在之后的连续N+1天里，每天投N+1张票。

请计算在前K天里，zyk一共获得了多少票。
 

输入

一个正整数K，表示投票的天数。

输出

一个正整数，即zyk收到的票数。

样例输入 Copy

6

样例输出 Copy

14

提示

【输入输出样例 1 说明】

zyk第一天收到一张选票；第二天和第三天，每天收到两张选票；第四、五、六天，每天收到三张选票。

因此，一共收到 1+2+2+3+3+3=14 枚金币。 对于 100%的数据，1 ≤ K ≤10,000。

提交

#include<iostream>
using namespace std;
int main()
{
    int n = 0, num = 0, i = 1,j = 0, t = 0, q = 0, p = 0;
    cin >> n;
    while(true)
    {
        q += i;
        p = q + i + 1;
        if( q < n && n < p )
        {
            t = n - q;
            for( j = 1; j <= i; j++ )
            {
                num += j*j;
            }
            cout << t*(i+1) + num;
            break;
        }
        else if( n == q || n == p )
        {
            for( j = 1; j <= (i+1); j++ )
            {
                num += j*j;
            }
            cout << num;
            break;
        }
        else
        {
            ++i;
        }
    }
    return 0;
}