1. 程式人生 > >題解 luogu P2757 【[國家集訓隊]等差子序列】

題解 luogu P2757 【[國家集訓隊]等差子序列】

首先,用一個桶儲存每個數的位置

因為等差數列只需三個便可成立,所以我們可以直接暴力列舉前兩個數,然後直接用桶判斷第三個數是否在前兩個數後面就可以了,直接輸出‘Y’

#include<bits/stdc++.h>
using namespace std;
int b[100100];
int a[100100];
int main() {
	int i,j,x,y,n,k,m,o;
	cin>>n;
	for (k=1; k<=n; k++) {
		cin>>x;
		memset(a,0,sizeof(a));
		memset(b,0,sizeof(b));
		o=0;
		for (i=1; i<=x; i++) {
			cin>>a[i];
			b[a[i]]=i;//做一個桶
		}
		for (i=1; i<=x; i++) {
			for (j=i+1; j<=x; j++) {
				if (b[a[j]-a[i]+a[j]]>j) {
					o=1;
					cout<<'Y'<<endl;
					break;//暴力列舉,然後看第三個數是否在前兩個數之前
				}
			}
			if (o==1)break;
		}
		if (o==0)cout<<'N'<<endl;
	}
	return 0;
}

結束!

一定要開O2!