#include<bits/stdc++.h> using namespace std; int n,k,a[1001]; int main(){ freopen("input.txt","r",stdin); freopen("output.txt","w",stdout); cin>>n>>k; for(int i=1;i<=n;i++){ cin>>a[i]; while(a[k]==0){ k++; if(k>n) k=1; } } cout<<k; }
一直TLE,大佬救救孩子吧