题意:
给我们一个长度为n(4<=n<=2e5)的数组a(0<=a[i]<=2.5e6),然后需要我们判断是否存在四个下标x,y,z,w,使得a[x]+a[y]=a[z]+a[w],存在则输出yes,否则no.
题解:
简单数论
其实这题就是两个for循环就可解决,看似n的数据范围为2e5,两个for循环一般来说肯定是超时的,然而,注意看a[i]的数据范围,只有2.5e6,那么对于2e5个数而言,当n的数据量比较大的时候,例如接近2e5时,那么在考虑a[i]的范围,这个时候就必然会出现接下来的一种会多种情况,存在四个a[i]相同,存在四个及以上个a[i]存在顺序关系(例如,1,2,3,4那么1+4=2+3,或者1,2,6,7这种1+7=2+6),那么其实就跑不满两个for,复杂度也就其实达不到 O(
n
2
n^{2}
n2),这题的数据量给的十分巧妙。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 200005;
int n,a[MAXN];
pair<int,int> b[30*MAXN];
int main(){
ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n;j++){
int sum=a[i]+a[j];
if(!b[sum].first) {b[sum]={i,j};continue;}
else if(i==b[sum].first || j==b[sum].first || i==b[sum].second || j==b[sum].second) continue;
cout<<"YES\n";
cout<<i<<" "<<j<<" "<<b[sum].first<<" "<<b[sum].second<<'\n';
return 0;
}
}
cout<<"NO\n";
return 0;
}