【劍指offer】連續子陣列的最大和

蘭亭風雨發表於2014-05-27

轉載請註明出處:http://blog.csdn.net/ns_code/article/details/27103959

題目描述:

HZ偶爾會拿些專業問題來忽悠那些非計算機專業的同學。今天JOBDU測試組開完會後,他又發話了:在古老的一維模式識別中,常常需要計算連續子向量的最大和,當向量全為正數的時候,問題很好解決。但是,如果向量中包含負數,是否應該包含某個負數,並期望旁邊的正數會彌補它呢?例如:{6,-3,-2,7,-15,1,2,2},連續子向量的最大和為8(從第0個開始,到第3個為止)。你會不會被他忽悠住?

輸入:

輸入有多組資料,每組測試資料包括兩行。

第一行為一個整數n(0<=n<=100000),n=0,輸入結束。接下去的一行包含n個整數(我們保證所有整數屬於[-1000,1000])

輸出:

對應每個測試案例,需要輸出3個整數單獨一行,分別表示連續子向量的最大和、該子向量的第一個元素的下標和最後一個元素的下標。若是存在多個子向量,則輸出起始元素下標最小的那個。

樣例輸入:
3
-1 -3 -2
5
-8 3 2 0 5
8
6 -3 -2 7 -15 1 2 2
0
樣例輸出:
-1 0 0
10 1 4
8 0 3
    前面有專門寫了一篇求最大連續子數個組的文章,見這裡:http://blog.csdn.net/ns_code/article/details/20942045。這裡的第三種做法,要在陣列中既有負數又有整數的情況下才能得到正確的結果。而這裡的測試樣例中有全負的情況,因此,跟上面的求解會有些細節上的不同,而且這裡除了要求出最大連續子陣列的和,還要求出該最大連續子陣列,因此要另設兩個變數儲存該連續子陣列的起始和結束的位置。

    AC程式碼如下:

#include<stdio.h>
#include<stdlib.h>
#include<stdbool.h>
bool flag;

int MaxSubSum(int *arr,int len,int *indexBegin,int *indexEnd)
{
	if(arr==NULL || len<1)
	{
		flag = false;
		return -1;
	}

	int i;
	int MaxSum;
	int CurSum;
	int CurBegin; //當前的CurSum的開始處的索引
	int CurEnd;	 //當前的CurSum的結束處的索引
	for(i=0;i<len;i++)
	{
		if(i == 0)
		{
			//先用第一個元素初始化
			*indexBegin = 0;
			*indexEnd = 0;
			CurBegin = 0;
			CurEnd = 0;
			MaxSum = arr[i];
			CurSum = arr[i];
		}
		else
		{
			//前面的和小於0,拋棄前面的和,從當前元素重新開始計算
			if(CurSum < 0)
			{
				CurSum = arr[i];
				CurBegin = i;
				CurEnd = i;
			}
			else
			{
				CurSum += arr[i];
				CurEnd = i;
			}
			if(CurSum > MaxSum)
			{
				MaxSum = CurSum;
				//如果當前和大於前面出現的最大和,
				//則替換掉最大和的開始索引和結束索引
				*indexBegin = CurBegin;
				*indexEnd = CurEnd;
			}
		}
	}
	return MaxSum;
}

int main()
{
	int n;
	while(scanf("%d",&n)!=EOF && n!=0)
	{
		int *arr = (int *)malloc(sizeof(int)*n);
		if(arr == NULL)
			exit(EXIT_FAILURE);

		int i;
		for(i=0;i<n;i++)
			scanf("%d",arr+i);

		flag = true;
		int indexBegin,indexEnd;
		int result = MaxSubSum(arr,n,&indexBegin,&indexEnd);	
		if(flag)
		{
			printf("%d ",result);
			printf("%d ",indexBegin);
			printf("%d\n",indexEnd);
		}

		free(arr);
		arr = NULL;
	}
	return 0;
}
/**************************************************************
    Problem: 1372
    User: mmc_maodun
    Language: C
    Result: Accepted
    Time:450 ms
    Memory:1304 kb
****************************************************************/

相關文章