簡體   English   中英

如何使用大陣列大小?

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM