金币(c++)

题目描述

国王将金币作为工资，发放给忠诚的骑士。第一天，骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天（第七、八、九、十天），每天收到四枚金币......；这种工资发放模式会一直这样延续下去：当连续 NN 天每天收到 NN 枚金币后，骑士会在之后的连续 N+1N+1 天里，每天收到 N+1N+1 枚金币。

请计算在前 KK 天里，骑士一共获得了多少金币。

输入描述

输入只有 1 行，包含一个正整数 K\ （1 \leq K \leq 10^4）K （1≤K≤104），表示发放金币的天数。

输出描述

输出只有 1 行，包含一个正整数，即骑士收到的金币数。

输入输出样例

示例 1

输入

6

输出

14

示例 2

输入

1000

输出

29820

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

#include<iostream>
using namespace std;

int main(){
	int n;
	cin>>n;
	int s=1;
	int w=1;
	long long sum=0; 
	for(int i=1;i<=n;i++){
		while(s--){
			if(i<=n){sum+=w;
			i++;
			}
			else {
				cout<<sum;
				return 0;
			}
		}
		w++;
		s=w;
		i--;
	}
	cout<<sum;
	return 0;
}