[英]How to use large array size?
我在spoj上嘗試這個問題。 它需要段樹,但問題在於數組的大小,這里(1 <= N <= 1,000,000,000)。有什么方法可以解決此問題? 這是我的實現(給出將近N <1000000的正確答案)
#include <stdio.h>
#include <math.h>
#include<iostream>
#include<string.h>
using namespace std;
//segment tree
long long a[1000000];
long long Mid(long long s,long long e)
{
return s+(e-s)/2;
}
long long Sum1(long long *st,long long ss,long long se,long long qs,long long qe,long long index)
{
if (qs<=ss&&qe>=se)
return st[index];
if (se<qs||ss>qe)
return 0;
long long mid=Mid(ss, se);
return Sum1(st,ss,mid,qs,qe,2*index+1) +Sum1(st,mid+1,se,qs,qe,2*index+2);
}
void update1(long long *st,long long ss,long long se,long long i,long long diff,long long index)
{
if (i<ss||i>se)
return;
st[index]=st[index]+diff;
if (se!=ss)
{
long long mid = Mid(ss,se);
update1(st,ss,mid,i,diff,2*index+1);
update1(st,mid+1,se,i,diff,2*index+2);
}
}
void update(long long arr[],long long *st,long long n,long long i,long long new_val)
{
if (i<0||i>n-1)
return;
long long diff = new_val - arr[i];
arr[i] = new_val;
update1(st,0,n-1,i,diff,0);
}
long long Sum(long long *st,long long n,long long qs,long long qe)
{
if (qs<0||qe>n-1||qs>qe)
return -1;
return Sum1(st,0,n-1,qs,qe,0);
}
long long segtree(long long arr[],long long ss,long long se,long long *st,long long si)
{
if (ss==se)
{
st[si]=arr[ss];
return arr[ss];
}
long long mid=Mid(ss, se);
st[si]=segtree(arr,ss,mid,st,si*2+1)+segtree(arr,mid+1,se,st,si*2+2);
return st[si];
}
long long *segt(long long arr[],long long n)
{
long long x = (long long)(ceil(log2(n)));
long long max_size = 2*(long long)pow(2, x) - 1;
long long *st = new long long[max_size];
segtree(arr,0,n-1,st,0);
return st;
}
int main()
{
//memset(a,0,sizeof(a));
long long n,u,v;
cin>>n>>u>>v;
for(long long i=0;i<n;i++)
a[i]=0;
long long *st=segt(a,n);
while(u--)
{
long long i,j;
cin>>i>>j;
long long z=1;
for(long long p=i-1;p<j;p++)
{
update(a,st,n,p,a[p]+z);
z++;
}
//for(int m=0;m<n;m++)
//cout<<a[m]<<endl;
}
while(v--)
{
long long i,j;
cin>>i>>j;
cout<<Sum(st,n,i-1,j-1)<<endl;
}
return 0;
}
在C或C ++中,本地對象通常或通常分配在堆棧上。 由於您要在堆棧上分配一個非常大的數組。 這樣您就有機會使堆棧溢出 。 我建議您使用std::vector<int>
並將其大小調整為1000000個元素。
無論您嘗試哪種解決方案,都將需要8 GB以上的ram才能使用此算法解決問題。 spoj的內存限制要少得多。 考慮一種需要較少內存的替代解決方案。
您可以嘗試使用binary indexed tree
而不是segment tree
-> 這是一個簡單的教程。
與段樹的O(2 ^(logN + 2))相比,BIT占用O(n)內存,並且可以達到相同的目的。
希望這可以幫助...
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.