循环数组求下标

给定包含N个整数的数组A1, A2, … AN,你可以选择任意一个Ai,将Ai旋转到数组第一项,即将数组变成:

Ai, Ai+1, Ai+2, … AN, A1, A2, …, Ai-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个整数A1, A2, … AN。

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

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

Output
一个整数表示答案。

Sample Input
6
3 -5 2 -2 3 0
Sample Output
5

题意不难理解,这题也不难写,暴力也能过,我一直没过是因为我数组开小了,真的该死,把n的值看漏了,这里暴力的就不写了,下面提供一中代码短,时间也少的方法,真的很nice

#include<stdio.h>
int a[1010];
int main()
{
    
    
	int n;
	while(~scanf("%d",&n))
	{
    
    
		int i,j=0,s=0;
		for(i=0;i<n;i++)
			scanf("%d",&a[i]);
		for(i=0;i<n-j;i++)
		{
    
    
			s+=a[i];
			while(s<=0)
			{
    
    
				j++;
				if(i<n-j)
					s+=a[n-j];
				else
					break;
			}
		}
		if(s>0)
			printf("%d\n",(n-j)%n+1);//这里记得对取模,因为会有n-j==n的情况
		else
			printf("-1\n");
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Helinshan/article/details/108919021