文章介绍了四个关于数组操作的问题，包括如何截断数组使各段和相等、如何改变数组元素、计算数组元素两两相乘的和以及找出数组中的K倍区间。每个问题都给出了具体的输入输出样例和解题思路，思路主要涉及前缀和、差分数组等算法，并讨论了如何优化暴力解决方法以应对大数据量的情况。
目录
1.截断数组
题目描述
给定一个长度为 n的数组 a1,a2,…,an。
现在，要将该数组从中间截断，得到三个非空子数组。
要求，三个子数组内各元素之和都相等。
请问，共有多少种不同的截断方法？
输入格式
第一行包含整数 n。
第二行包含 n 个整数 a1,a2,…,an。
输出格式
输出一个整数，表示截断方法数量。
数据范围
前六个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤1e5，−10000≤ai≤10000。
输入样例1
4
1 2 3 3
输出样例1
1
输入样例2
5
1 2 3 4 5
输出样例2
0
输入样例3
2
0 0
输出样例3
0思路
利用前缀和思想:
分别用两个指针j,i表示要切掉的中间的两刀，由于分成三个相等的部分，每部分之和都是总和
的三分之一才满足条件，如果满足条件，那么sum%3=0,先用i指针表示所要切的第二刀，
那么sum[i,n]==sum/3,再枚举j满足sum[1,j]==sum/3的条件的个数，由于枚举i指针的同时不
影响j指针，那么可以同时枚举i，j指针,因为分成3个部分，那么i从3开始枚举。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long LL;
const int N=1e5+10;
int s[N];
int n;
LL res;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>s[i];
s[i]+=s[i-1];
}
if(s[n]%3) puts("0");
else
{
for(int i=3,cnt=0;i<=n;i++)
{
if(s[i-2]==s[n]/3) cnt++;
if((s[n]-s[i-1]==s[n]/3)) res+=cnt;
}
cout<<res<<endl;
}
return 0;
}2.改变数组元素
题目描述
给定一个空数组 V 和一个整数数组 a1,a2,…,an。
现在要对数组 V进行 n次操作。
第 i 次操作的具体流程如下：
- 从数组 V尾部插入整数 0。
- 将位于数组 V末尾的 ai个元素都变为 1（已经是 1 的不予理会）。
注意：
- ai可能为 0，即不做任何改变。
- ai可能大于目前数组 V所包含的元素个数，此时视为将数组内所有元素变为 1。
请你输出所有操作完成后的数组 V。
输入格式
第一行包含整数 T，表示共有 T 组测试数据。
每组数据第一行包含整数 n。
第二行包含 n个整数 a1,a2,…,an。
输出格式
每组数据输出一行结果，表示所有操作完成后的数组 V�，数组内元素之间用空格隔开。
数据范围
1≤T≤20000,
1≤n≤2×1e5,
0≤ai≤n,
保证一个测试点内所有 n的和不超过 2×1e5。
输入样例
3
6
0 3 0 0 1 3
10
0 0 0 1 0 5 0 0 0 2
3
0 0 0
输出样例
1 1 0 1 1 1
0 1 1 1 1 1 0 0 1 1
0 0 0思路
差分思想:
由于每次都是插入一个数0，并且把后a[i]个数变成1，但是我们并不需要真的每次都把
后几个位置变成1，每次只需要在后几个位置上面+1即可，因为这样的话最后哪个位置
上面不为0的点，就是需要变成1的点，可以利用差分数组的思想，最后，构造差分数组
的前缀和数组，如果构造的前缀和数组非0，那么输出1，如果为0，则输出0，输入输出
用scanf和printf.
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 2e5+10;
int a[N],n,T,v[N],d[N];
int main()
{
cin>>T;
while(T--)
{
memset(d,0,sizeof d);
memset(v,0,sizeof v);
int tt=0;
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
v[++tt]=0;//尾部插入0
int l=tt-a[i]+1;
if(l<=1) l=1;
d[l]+=1;
d[tt+1]-=1;
}
for(int i=1;i<=n;i++)
{
v[i]=v[i-1]+d[i];
if(v[i]) printf("1 ");
else printf("0 ");
}
puts("");
}
return 0;
}3.求和
题目描述
给定 n个整数 a1,a2,⋅⋅⋅,an，求它们两两相乘再相加的和，即
S=a1⋅a2+a1⋅a3+⋅⋅⋅+a1⋅an+a2⋅a3+⋅⋅⋅+an−2⋅an−1+an−2⋅an+an−1⋅an
输入格式
输入的第一行包含一个整数 n。
第二行包含 n个整数 a1,a2,⋅⋅⋅,an。
输出格式
输出一个整数 S，表示所求的和。
请使用合适的数据类型进行运算。
数据范围
对于 30%30% 的数据，1≤n≤1000，1≤ai≤100。
对于所有评测用例，1≤n≤200000，1≤ai≤1000。
输入样例
4
1 3 6 9
输出样例
117思路
每一项都添加一些项，最后减去这些项：
只需要把式子做一个变形，把a1*(a2+a3+..+an)+a2*(a3+a4+...+an)+...
变形为a1*(a1+a2+...+an)+a2(a1+a2+...+an)+...-a1*a1-a2*(a1+a2)-...
设sn为an的前缀和数组，则上面的式子可以变形为
a1*sn+a2*sn+...+an*sn-a1*s1-a2*s2-a3*s3-...-an*sn.
再化简sn*sn-a1*s1-a2*s2-...-an*sn.
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200010;
typedef long long ll;
ll a[N],s[N];
int n;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s[i]=s[i-1]+a[i];
}
ll res=s[n]*s[n];
ll ans=0;
for(int i=1;i<=n;i++)
ans+=a[i]*s[i];
cout<<ll(res-ans)<<endl;
return 0;
}4.k倍区间
题目描述
给定一个长度为 NN 的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗？
输入格式
第一行包含两个整数 N 和 K。
以下 N 行每行包含一个整数 Ai。
输出格式
输出一个整数，代表 K 倍区间的数目。
数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例
5 2
1
2
3
4
5
输出样例
6
思路
找到一个区间[l,r],使得[l,r]区间的所有的数的和模k为0，那么可以构造前缀和数组s,
则(s[r]-s[l-1])%k==0，
暴力做法
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
if((s[i]-s[j-1]))%k==0) ans++;
但是由于题目中的数据是1e5，所以需要对暴力方法做优化。
对于一般的(s[r]-s[l-1])%k==0,可以转化成为s[r]%k=s[l-1]%k,
对于每个s[i],只需要找到s[i]%k的数的个数，用一个cnt数组存储s[i]%k的数的个数。
for(int i=1;i<=n;i++)
{
ans+=cnt[s[i]%k];//找到符合s[i]%k数的个数
cnt[s[i]%k]++;
}
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long LL;
const int N = 1e5+10;
int n,k;
LL a[N],s[N],cnt[N];
LL ans;
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s[i]=s[i-1]+a[i];
}
cnt[0]=1;
for(int i=1;i<=n;i++)
{
ans+=cnt[s[i]%k];
cnt[s[i]%k]++;
}
cout<<ans<<endl;
return 0;
}
3272
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
