題目鏈接
分析
這道題巧妙的應用了hash和邊表;
首先分析一下題意,就是讓你在n個數裏找3個數加起來等於第四個數。
ai+aj+ap=ak <==> ai+aj=ak-ap 之後就簡單了就是邊表的技巧
代碼
#include<bits/stdc++.h>
#define N 12600000
#define M 4194303
using namespace std;
int n,a[5005];
int tot,head[N],Next[N],val[N];
void add(int x){
tot++;
val[tot]=x;
Next[tot]=head[x & M];
head[x & M]=tot;
}//x & M 就是hash的過程
bool check(int w){
for(int i=head[w & M];i;i=Next[i])
if(val[i]==w)return true;
return false;
}//判斷是否有答案
int Hash(){
int ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<i;j++)
if(check(a[i]-a[j])){
ans++;
break;
}
for(int j=1;j<=i;j++)
add(a[i]+a[j]);
}//大暴力
return ans;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
printf("%d",Hash());
return 0;
}//^_^畫一個笑臉表示自信