input:
4
1 2 3 3
output:
1
看到这个题第一眼想到了前缀和,然后两重循环去做判断是否能够满足我们的要求,但是数据范围是100000,O(N*N)肯定是过不了的,因此我们要去寻求优化。
其具体的思路如下:
因为要把数组均分三份,总和一定得是3的倍数
遍历前缀和数组,边扫描边记录哪些地方可以切第一刀,哪些地方可以切第二刀。如果位置j可以切第二刀,那么它与前面第一刀进行组合,就可以切成三部分。
切第一刀的判断条件是:s[i]==t;
处理方法是:cnt++(cnt的意思就是i位置前面有多少可以切第一刀的地方)
切第二刀的判断条件是:s[n]-s[i+1]==t
处理方法是:ans+=cnt,第二刀跟前面的cnt个第一刀组合,所以总个数增加cnt个
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define endl "\n";
const int N = 1e5 + 10;
int a[N], q[N];
int main()
{
int n;
ll sum = 0;
cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
sum += a[i];
q[i] = q[i - 1] + a[i];//初始化前缀和
}
if (sum % 3)//如果不能分成三个部分,直接就不用判断了
{
cout << 0 << endl;
}
else
{
ll res = 0, ave = sum / 3;
for (int i = 2, cnt = 0; i <= n - 1; i++)//尽量去枚举后面的一个点,这是大佬的经验
{
if (q[i - 1] == sum / 3)
cnt++;
if (q[i] == 2 * ave)//如果第一刀满足条件,并且第二刀也满足条件,将两个组合起来就是答案
res += cnt;
}
cout << res << endl;
}
return 0;
}
1230. K倍区间(活动 - AcWing)
input:
5 2
1 2 3 4 5
output:
6
我们和容易想到前缀和和枚举去做这个题,但是如果两重循环去枚举的话,时间肯定是会超限的。那么这个题该如何去优化。
具体思路:
s[i]=a[0]+a[1]+a[2]+a[3]+a[4]+a[5]+...........+a[i-1]+a[i]
满足要求的前缀和是
(s[i]-s[j1] )%k==0
(s[i]-s[j2] )%k==0
(s[i]-s[j3] )%k==0
通过简单的变换我们可以得到的:s[i]%k==s[j]%k
所以说我们可以通过枚举当前这个点前的和我余数相同的个数有多少,答案直接加上就可以了,注意我们在枚举的时候%k==0的点要特殊处理一下。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll a[N], s[N], cnt[N];
int main()
{
ll n, k, res = 0;
cin >> n >> k;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
s[i] = s[i - 1] + a[i];//初始化前缀和
}
cnt[0] = 1;//这个就是处理%k==0的情况
for (int i = 1; i <= n; i++)
{
res += (cnt[s[i] % k]);//答案加上和我余数相同的个数
cnt[s[i] % k]++;//余数的处理
}
cout << res << endl;
return 0;
}
这个题目是二维前缀和的使用的理解
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5010;
int a[N][N], s[N][N];
int n, m;
int main()
{
cin >> n >> m;
m = min(m, 5001); // r的范围太大了,提前处理一下
for (int i = 0; i < n; i++)
{
int x, y, w;
cin >> x >> y >> w;
s[x + 1][y + 1] += w; // 每个点可能会有多个价值叠加在一起
}
for (int i = 1; i <= 5001; i++)
for (int j = 1; j <= 5001; j++)
s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
// 求前缀和数组
int res = 0;
for (int i = m; i <= 5001; i++)
for (int j = m; j <= 5001; j++)
res = max(res, s[i][j] - s[i - m][j] - s[i][j - m] + s[i - m][j - m]);
// 依次枚举每一个小正方形的价值,取得结果
cout << res << endl;
return 0;
}