求助,最后一个点TLE
查看原帖
求助,最后一个点TLE
175149
Bean233楼主2023/4/14 16:47
#include <iostream>
#include <cstdio>
using namespace std;
int main()
{
	int n,m;
	cin >> n >> m;
	int a[n+100];
	for(int i=1;i<=n;i++) {
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=m;i++) {
		int q;//1 1 2 3 5
		scanf("%d",&q); //3
		int l=1,r=n,mid=(l+n)/2;
		while(l<r) {
			int p=0;
			if(q==a[l]) {
				cout << l << " ";
				break;
			}else if(q==a[r]) {
				for(int i=r;i>=1;i--) {
					if(a[i]!=q) {
						cout << i+1 << " ";
						p=1;
						break;
					}
				}
				break;
			} else if(q>a[mid]) {
				l = mid;
			}else if(q<a[mid]) {
				r = mid;
			}else if(q==a[mid]) {
				for(int i=mid;i>=1;i--) {
					if(a[i]!=q) {
						cout << i+1 << " ";
						p=1;
						break;
					}
				}
			}
			if(p==1) {
				break;
			}
			mid = (l+r)/2;
			if(mid==r||mid==l) {
				cout << -1 << " ";
				break;
			}
		}
	}
	
	return 0;
}

RT

2023/4/14 16:47
加载中...