（算法）和为0的最大连续子数组

#include<iostream>
#include<vector>
#include<map>

using namespace std;

// O(n^2)
int longestSubArrayOfSumZero_1(const vector<int> &arr){
    int sz=arr.size();
    vector<int> preSum(sz+1,0);

    for(int i=0;i<sz;i++){
        preSum[i+1]=preSum[i]+arr[i];
    }

    int longest=0;
    int start=0;
    for(int i=1;i<=sz;i++){
        for(int j=0;j<i;j++){
            if(preSum[i]==preSum[j] && (i-j)>longest){
                longest=i-j;
                start=j;
            }
        }
    }
    
    for(int i=start;i<start+longest;i++)
        cout<<arr[i]<<" ";
    cout<<endl;

    return longest;
}


// O(n)
int longestSubArrayOfSumZero_2(const vector<int> &arr){
    int sz=arr.size();
    int longest=0;
    map<int,int> pos;
    int sum=0;
    pos[sum]=0;
    int start=0;

    for(int i=0;i<sz;i++){
        sum+=arr[i];
        if(pos.find(sum)!=pos.end()){
            int len=i-pos[sum];
            if(len>longest){
                longest=len;
                start=pos[sum]+1;
            }
        }
        else
            pos[sum]=i;

    }

    for(int i=start;i<start+longest;i++)
        cout<<arr[i]<<" ";
    cout<<endl;
    return longest;
}


int main(){
    int n;
    while(cin>>n){
        vector<int> arr(n);
        for(int i=0;i<n;i++){
            cin>>arr[i];
        }

        cout<< longestSubArrayOfSumZero_1(arr) <<endl;
        cout<< longestSubArrayOfSumZero_2(arr) <<endl;
    }
    return 0;
}