Question

我在n个元素上有一个数组（1＆lt; = n＆lt; = 200000）。我想要找到可以从这个数组形成的每个重叠子阵列的总和。我有一个O（n ^ 2）算法可以找到所有的总和，但我的问题是我不能将它存储在任何数据结构中，因为有n（n + 1）/ 2个元素。因此将有10 ^ 10个元素，这将需要大的空间。这是我得到的输出。

terminate called after throwing an instance of 'std::bad_alloc' what(): std::bad_alloc Aborted (core dumped)

我想是因为我的代码使用了太多内存。以下是我的代码。

#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
#define vi vector<int>
#define vli vector<lli>
#define dqi deque<int>
#define MOD 10e9+7
#define mis map<int,string>
#define msi map<string,int>
#define set0(a) memset(a,0,sizeof(a))
#define sc scanf
#define pr printf
#define rint(a) sc("%d",&a)
#define rchar(a) sc("%d",&a)
#define pb push_back
#define pf push_front
#define rstring(s) sc("%s",&s)
#define rp(a,b,c) for(int (a)=(b);(a)<(c);(a)++)
#define rpn(a)  while(a--)
int a[200010],t=0,n=0,q=0,cnt=0;
vector<long long int> b;
long long int l=0,r=0;
int main()
{
  freopen("in.txt","r",stdin);
  freopen("out.txt","w",stdout);
  memset(a,0,sizeof(a));
  scanf("%d",&t);
  while(t--){
    scanf("%d %d",&n,&q);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    long long int sum=0;
    for(int i=n-1,j=1;i>=0;i--,j++){
        b.push_back(a[i]);sum=a[i];
        for(int j=i+1;j<=n-1;j++)
                b.push_back(sum+=a[j]);
    }
    //following part find the sum of elements of subarrays in a given range after sorting them
    printf("Case #%d: \n",++cnt);
    while(q--){
        scanf("%lld %lld",&l,&r);
        long long int sum=0;
        for(long long int i=l-1;i<r;i++)
            sum+=b[i];
        printf("%lld\n",sum);
    }
    b.clear();
}
return 0;
}

还有其他方法可以做到这一点。请指导。

Answer 1

您可以使用通常用于此类问题的分段树。浏览下面给出的链接

https://www.hackerearth.com/practice/notes/segment-tree-and-lazy-propagation/

不是给出两个元素的差异（在链接中给出），而是可以在根中存储元素的总和。希望它有所帮助。

如何存储数组的每个重叠子阵列的总和？

1 个答案: