Я пытался эту проблему на спой. www.spoj.com/problems/RRANGE. Требуется дерево сегментов. Но проблема в размере массива. Здесь (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 ГБ оперативной памяти для решения проблемы с помощью этого алгоритма. Ограничение памяти на спой намного меньше. Подумайте об альтернативном решении, которое требует меньше памяти.
Вы можете попробовать использовать binary indexed tree
вместо segment tree
-> Вот это нативный учебник.
BIT занимает O (n) памяти по сравнению с O (2 ^ (logN + 2)) дерева сегментов и может служить той же цели.
Надеюсь это поможет…