P2669 [NOIP2015 普及组] 金币

#include<iostream>
using namespace std;
int main()
{
	int n,x=1,c=1,t=0;
	int sn[1005] = { 0 };
	cin >> n ;
	for (int i = 1;; ++i)
	{
		sn[i] =sn[i-1]+ i;//下标存一天拿的金币，数组里存那拿下标个金币的最后的天数
		//sn[i + 1] = sn[i] + 1 + i;
		if (sn[i] == n)
		{
			t += (i * i);
			break;
		}
		if (sn[i-1] < n&&sn[i]>n)
		{
			t += ((n-sn[i-1]) * i);
			break;
		}
		t += (i * i);
	}
	//while (n--)
	//{
	//	t += c;
	//	x--;//那这笔钱的天数
	//	if (x == 0)
	//	{
	//		c++;
	//		x = c;
	//	}
	//}
	cout <<t;
	return 0;
}