#include<bits/stdc++.h> using namespace std; int main(){ int n; cin>>n; vector<int> v(n); for(int i=0;i<n;i++){ cin>>v[i]; } long long num=0; unordered_map<int,int> rec; for(int i=0;i<n-2;i++){ rec.clear(); rec[v[i+1]]++; for(int k=i+2;k<n;k++){ if((v[i]+v[k])%3 !=0 ){ rec[v[k]]++; continue; } int tar=(v[i]+v[k])/3; num+=rec[tar]; rec[v[k]]++; } } cout<<num<<endl; } 这样可以过

相关推荐

牛客网
牛客企业服务