迴圈陣列最大子段和

brucehb發表於2017-06-28
陣列
N個整陣列成的迴圈序列a[1],a[2],a[3],…,a[n]，求該序列如a[i]+a[i+1]+…+a[j]的連續的子段和的最大值（迴圈序列是指n個數圍成一個圈，因此需要考慮a[n-1],a[n],a[1],a[2]這樣的序列）。當所給的整數均為負數時和為0。
例如：-2,11,-4,13,-5,-2，和最大的子段為：11,-4,13。和為20。
Input
第1行：整數序列的長度N（2 <= N <= 50000)
第2 - N+1行：N個整數 (-10^9 <= S[i] <= 10^9)
Output
輸出迴圈陣列的最大子段和。
Input示例
6
-2
11
-4
13
-5
-2
Output示例
20

#include <iostream>
#include <algorithm>
using namespace std;

int input[50000];

long long int maxSum(int n)
{
	long long int result = input[0];
	long long int sum = input[0];

	for (int i = 1; i < n; i++)
	{
		if (sum < 0)
		{
			sum = input[i];
		}
		else
		{
			sum += input[i];
		}

		if (sum > result)
		{
			result = sum;
		}
	}

	return result;
}

long long int minSum(int n)
{
	long long int result = input[0];
	long long int sum = input[0];

	for (int i = 1; i < n; i++)
	{
		if (sum > 0)
		{
			sum = input[i];
		}
		else
		{
			sum += input[i];
		}

		if (sum < result)
		{
			result = sum;
		}
	}
	
	return result;
}

long long int totalSum(int n)
{
	long long int result = 0;
	for (int i = 0; i < n; i++)
	{
		result += input[i];
	}

	return result;
}

int main(int argc, const char * argv[])
{
	int n;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> input[i];
	}

	long long sum1 = maxSum(n);
	long long sum2 = totalSum(n) - minSum(n);
	cout << max(sum1, sum2) << endl;

	return 0;
}


相關文章