小雨能找到一张好座椅吗？

目录

输入格式

输出格式

不知道怎么的，小雨来到了一个奇怪的公园，看到一些奇怪的座椅。这些座椅编号从1到n，以环形排列，也就是说，按顺时针顺序，1号座椅的下一个是2号，2号座椅的下一个是3号，n号座椅的下一个是1号。

除了按顺时针顺序编号之外，每个座椅还有一个数字，表示这个座椅上已经坐了几个人，只有一个人没有的座椅，小雨才觉得是好座椅，才愿意坐下去。

现在已经知道小雨从第几号座椅出发，按照顺时针顺序挨个经过座椅，请你编程计算小雨在找到合适座椅的过程中要经过几个座椅。

输入格式

第一行两个整数，分别是n（1≤n≤1000）和小雨从第几号座椅的位置开始出发。

第二行n个整数，依次表示每个座椅已经坐了几个人。

输出格式

一个整数，表示小雨经过几个座椅后能找到合适的坐下，如果小雨始终找不到合适的，则输出0。

 

---

 

#include <iostream>
using namespace std;
int a[10005];
int main ()
{
	int n,b;
	cin>>n>>b;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	int flag=0;
	for(int i=b;i<n+b;i=(i+1)%n)
	{
		flag++;
		if(a[i]==0)
		{
			cout<<flag;
			break;
		}
		if(flag>n)
		{
			cout<<0;
		}
	}
    return 0;
}