pat乙级1120(买地攻略) C++

算法初学者，第一次写解析，表达能力有限，望大佬指点。

第一是想的使用暴力求解，但是粗略心算了一下时间复杂度，感觉是会超时。于是想到了前缀和。

1.定义了一个存储价格的数组，然后sum用来存储前缀和。

2.从第一块田地开始遍历，在第i块田地购买的方案有两种可能性：i - k 或者 i - k + 1。

在买田时 ：sum(k, i) <= M(拥有的现金)，k是开始买的地的下标，i是遍历到的那块地的下标。

3.最后只需要将在每块地购买的种类数相加即可。

#include 
#define IOScc ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
using namespace std;
typedef long long ll;
int arr[10100];
ll sum[10100];

int main()
{
	IOScc;
	arr[0] = sum[0] = {0};
	int n, m, ans = 0;
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
	{
		cin >> arr[i];
		sum[i] = arr[i] + sum[i - 1];    // 前缀和
	}
	for(int i = 1; i <= n; i++)
	{
		ll s = sum[i], k = 0;
		while(s > m)
		{
		    s = sum[i];
			s = s - sum[k];    // 求sum(k,i)的价格, 以此确定要从哪块地开始买
			k++;
		}
		if(k)    // 讨论购买田地的可能性
			ans += (i - k + 1);
		else
		    ans += (i - k);
	}
	cout << ans;
	return 0;
}