C0224 [2015普及组-A]金币-C语言写

题目描述

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

请计算在前K天里，骑士一共获得了多少金币。
输入描述

只有1行，包含一个正整数K，表示发放金币的天数。

输出描述

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

样例输入 1

6

样例输出 1

14

样例输入 2

1000

样例输出 2

29820

代码：

#include
#include
void salary(int K);
int main(void){
	int K;
	scanf("%d",&K);//输入K 
	salary(K);
	return 0;
}
void salary(int K){
	int sum=0,money=0,sum_prior=0;
	for(int i=1;i<=K;i++){
		sum = sum+i; //求最小的前n项和（大于K） 
		sum_prior = sum_prior+(i-1);//求前n-1项和 
		money = money+pow((i-1),2); 
		if(sum==K){
			money = money+pow(i,2);
			printf("%d\n",money);
			break;
		}
		if(sum>K){
			money = money+(K-sum_prior)*i;
			printf("%d\n",money);
			break;
		}
	}
}