上海市计算机学会竞赛平台2024年7月月赛丙组求和问题

题目描述

给定 nn 个整数 a1,a2,…,ana1​,a2​,…,an​，请问这个序列最长有多少长的前缀，满足元素的和大于或等于 00？如果任何长度大于 00 的前缀之和都为负数，则输出 00

输入格式

• 第一行：单个整数表示 nn
• 第二行：nn 个整数表示 a1,a2,…,ana1​,a2​,…,an​

输出格式

• 单个整数：表示最长的前缀长度，使得前缀的和大于等于 00

数据范围

• 30%30% 的数据，1≤n≤1001≤n≤100
• 60%60% 的数据，1≤n≤10,0001≤n≤10,000
• 100%100% 的数据，1≤n≤200,0001≤n≤200,000
• −1,0000≤ai≤1,0000−1,0000≤ai​≤1,0000

样例数据

输入:

3
1 2 3

输出:

3

输入:

5
1 2 -1 3 -6

输出:

4

详见代码:

#include<bits/stdc++.h>
using namespace std;
int n;
int a[200005];
int ans = 0;
int main() 
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] += a[i - 1];
        if (a[i] >= 0) 
        {
            ans = i;
        }
    }
    cout << ans;
    return 0;
}