输入:int 型数组
输出:最长和为0的子序列
例:
输入:[3,0,-1,-2,-3,1,1,1,2,3,1,-2,-1]
输出:9
解题思路:
新建一个数组,sumvec,sumvec[0] = 0,sumvec[i]=a[0]+a[1]+……+a[i-1]
问题转化为求数组两个相同数字最远距离,之所以令 sumvec[0] = 0,是因为数组中从第一个元素到第 i 个元素和为 0 时,第一个元素也为 0,这样可以找到从第一个开始的最长子序列。
得到 sumvec 后,查找相同的距离最远的元素,每碰见一个新元素,元素值为key,下标为 value 放入 unordered_map 中,如果该元素已经出现了,则找到了和为 0 的子序列,如果长度大于前面的字串,更新长度,首尾位置。
时间复杂度:O(n)
空间复杂度:O(n)
#include <iostream>
#include <vector>
#include<unordered_map>
using namespace std;
//a为待求解的数组,res为求解的和为0的最长字串
int longestnum(vector<int>a, vector<int>&res){
vector<int>sumvec{0}; //和数组,第一个元素为0
int sum = 0;
for (int c : a){
sum += c;
sumvec.push_back(sum);
}
unordered_map<int, int>m;
int l = 0, r = -1, len = 0;
for (int i = 0; i < sumvec.size(); i++){
auto it = m.find(sumvec[i]);
if (it != m.end()){ //元素出现过,找到了和为0的子数列
if (i - it->second > len){
len = i - it->second; //更新长度
l = it->second; //更新左侧位置
r = i -1; //更新右侧位置
}
}
else
m[sumvec[i]] = i; //元素没有出现过,插入新元素
}
for (int i = l; i <= r; i++)
res.push_back(a[i]);
return len;
}
int main() {
vector<int>vec{ 3, 0, -1, -2, -3, 1, 1, 1, 2, 3, 1, -2 };
vector<int>res;
cout << longestnum(vec, res) << endl;
for (int c : res)
cout << c << " ";
return 0;
}