Maximum Subsequence Sum【最大连续子序列+树状数组解决】

Problem Description
给定K个整数的序列{ N1, N2, ..., NK },其任意连续子序列可表示为{ Ni, Ni+1, ..., Nj },其中 1 <= i <= j <= K。最大连续子序列是所有连续子序列中元素和最大的一个, 例如给定序列{ -2, 11, -4, 13, -5, -2 },其最大连续子序列为{ 11, -4, 13 },最大和 为20。 在今年的数据结构考卷中,要求编写程序得到最大和,现在增加一个要求,即还需要输出该 子序列的第一个和最后一个元素。
 
Input
测试输入包含若干测试用例,每个测试用例占2行,第1行给出正整数K( < 10000 ),第2行给出K个整数,中间用空格分隔。
 
Output
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元 素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。
 
Sample Input
6
-2 11 -4 13 -5 -2
 
10
-10 1 2 3 4 -5 -23 3 7 -21
 
6
5 -8 3 2 5 0
 
1
10
 
3
-1 -5 -2
 
3 -
1 0 -2
 
Sample Output
20 11 13
10 1 4
10 3 5
10 10 10
0 -1 -2
0 0 0
 
我用树状数组写的
code:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <algorithm>

using namespace std;

int lowbit(int x)
{
    return x&(-x);
}

int a[100000+10];
int c[100000+10];
int s[100000+10];

int Sum(int x)
{
    int s=0;
    while(x>0)
    {
        s+=c[x];
        x=x-lowbit(x);
    }
    return s;
}

bool fu(int *a, int n)
{
    for(int i=1; i<=n; i++)
    {
        if(a[i]>=0) return false;
    }
    return true;
}
int main()
{
    int n;
    int i, j;
    while(scanf("%d", &n)!=EOF)
    {
        for(i=1; i<=n; i++)
        {
            scanf("%d", &a[i]);
        }
        if(fu(a, n)==true)
        {
            printf("0 %d %d\n", a[1], a[n]);
            continue;
        }
        for(i=1; i<=n; i++)
        {
            c[i]=0;
            for(j=i-lowbit(i)+1; j<=i; j++)
                c[i]+=a[j];
            //printf("c[%d]=%d\n", i, c[i]);
        }
        int ans=-1;
        for(i=1; i<=n; i++)
            s[i]=Sum(i);

        int left, right;
        for(i=1; i<=n; i++)
        {
            int dd=s[i], ff=0;
            if(dd>ans)
            {
                ans=dd;
                left=a[1];
                right=a[i];
            }
            for(j=1; j<i; j++)
            {
                ff=s[j];
                if(dd-ff > ans)
                {
                    ans = dd-ff;
                    left=a[j+1];
                    right=a[i];
                }
            }
        }
        printf("%d %d %d\n", ans, left, right);
    }
    return 0;
}

 

posted @ 2015-09-06 09:42  我喜欢旅行  阅读(379)  评论(0编辑  收藏  举报