C. Dominant Piranha(思維)

2020-10-22 11:00:47

https://codeforces.com/contest/1433/problem/C


思路:找序列中最大的,如果滿足有一個最大的數旁邊存在比它小的數,那麼必然能YES,輸出這個數位置即可。如果不能滿足就是NO。

 

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=3e5+100;
typedef long long LL;
LL a[maxn];
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL t;cin>>t;
  while(t--)
  {
  	LL n;cin>>n;
  	for(LL i=0;i<=n+10;i++) a[i]=0; 
  	for(LL i=1;i<=n;i++) cin>>a[i];
  	LL mx=-1e18;
  	for(LL i=1;i<=n;i++){
  		if(mx<a[i]) mx=a[i];	
	}
	//LL sum=0;
	LL pos1=0;
	for(LL i=1;i<=n;i++){
		if(mx==a[i]) {
			//if(((i-1)!=0&&a[i]>a[i-1])&&(a[i]>a[i+1]&&(i+1)!=n))
		//	{
			//	pos1=i;break;
			//}	
			if(i==1&&a[1]>a[2]){
				pos1=1;break;
			}
			else if(i==n&&a[n]>a[n-1]){
				pos1=n;break;
			}
			else if( ((i!=1)&&a[i]>a[i-1])||(i!=n&&a[i]>a[i+1])){
				pos1=i;break;
			}
		}
	}
	if(pos1==0){
		cout<<"-1"<<endl;
	}
	else cout<<pos1<<endl;
  }
return 0;
}