循环数组 HihoCoder - 1704

循环数组

  HihoCoder - 1704 

给定包含N个整数的数组A₁, A₂, ... A_N，你可以选择任意一个A_i，将A_i旋转到数组第一项，即将数组变成：

A_i, A_i+1, A_i+2, ... A_N, A₁, A₂, ..., A_i-1  

现在小Hi希望旋转之后的数组满足：

对于任意K(1 ≤ i ≤ N)，前K项的和都是正数。  

例如对于A=[3, -5, 2, -2, 3, 0]，旋转成[3, 0, 3, -5, 2, -2]满足条件。  

请你输出i，代表将Ai旋转到第一项满足条件。  

如果有多解，你可以输出任意一个i。如果无解输出-1。

Input

第一行包含一个整数N。  

第二行包含N个整数A₁, A₂, ... A_N。  

对于50%的数据，1 ≤ N ≤ 1000  

对于100%的数据，1 ≤ N ≤ 100000, -1000000 ≤ A_i ≤ 1000000

Output

一个整数表示答案。

Sample Input

6  
3 -5 2 -2 3 0

Sample Output

5

代码：

#include
#include
#include
using namespace std;
#define N 100005
int a[N];
int main()
{
    int n;
    long long ans=0;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        ans+=a[i];
    }
    if(ans<=0)
        printf("-1\n");
    else
    {
        int h,flag=0;
        long long k;
        for(int i=1; i<=n; i++)
        {
            k=0;
            flag=0;
            for(int j=i;jn)
                    h=j%n;
                else
                    h=j;
                k+=a[h];
                if(k<=0)
                {
                    flag=1;
                    break;
                }
            }
            if(!flag)
            {
                printf("%d\n",i);
                break;
            }
        }
        if(flag)
            printf("-1\n");
    }
     return 0;
}