c++ - 如何修改 Kadane 的算法以找出对最大和有贡献的子数组元素？

Question

到目前为止我所做的：

我用 Kadane 的算法找到了总和。

我创建了一个向量“res”，它在满足条件时存储元素。

问题： 例如，当输入是一个数组时： {-2, -3, 4, -1, -2, 1, 5, -3} 结果数组包含右子数组元素以及添加的 1 个额外元素当条件失败时。

tldr：代码将输出子数组返回为：{4 -1 -2 1 5 -3} 而正确的输出应该是 {4, -1, -2, 1, 5}

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int i, a[10], n, temp_sum=0, final_sum=0;
    vector<int> res;
    cin >> n;                               //getting array size input
    for(i=0; i<n; i++)
    {
        cin >> a[i];                        //getting array elements input
    }
    temp_sum=a[0];                          //initializing to first element
    final_sum=a[0];
    for(i=1; i<n; i++)
    {
        if(a[i]>temp_sum+a[i])       //if element is greater than sum so far
        {
            temp_sum = a[i];
            res.clear();
            res.push_back(a[i]);
        }
        else
        {
            temp_sum = a[i] + temp_sum;
            cout << temp_sum << " " << final_sum << " " << a[i] << "\n";
            res.push_back(a[i]);
        }
        if(temp_sum>final_sum)
        {
            final_sum = temp_sum;
        }
    }
    cout << final_sum << "\n";                  //displaying maximum sum
    for(i=0; i<res.size(); i++)                 //displaying result vector
    {
        cout << res[i] << " ";
    }
    return 0;
}

Answer 1

请注意，您执行

res.push_back(a[i]);

if(temp_sum>final_sum)在任何情况下，如果失败也不刷新 res

顺便说一句，条件if(a[i]>temp_sum+a[i])等于 temp_sum <0 （不影响结果）