算法初学者,第一次写解析,表达能力有限,望大佬指点。
第一是想的使用暴力求解,但是粗略心算了一下时间复杂度,感觉是会超时。于是想到了前缀和。
1.定义了一个存储价格的数组,然后sum用来存储前缀和。
2.从第一块田地开始遍历,在第i块田地购买的方案有两种可能性:i - k 或者 i - k + 1。
在买田时 :sum(k, i) <= M(拥有的现金),k是开始买的地的下标,i是遍历到的那块地的下标。
3.最后只需要将在每块地购买的种类数相加即可。
#include <iostream>
#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;
}