fork download
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int main()
  4. {
  5. int i,j,k,n;
  6. cin>>n;
  7. int arr[n];
  8. vector<int> lhs,rhs;
  9. vector<pair<int,int>> nlhs,nrhs;
  10. for(i=0;i<n;i++){
  11. cin>>arr[i];
  12. }
  13.  
  14. for(i=0; i<n; i++)
  15. for(j=0; j<n; j++)
  16. for(k=0; k<n; k++){
  17. if(arr[i])
  18. rhs.push_back(arr[i]*(arr[j]+arr[k]));
  19. lhs.push_back((arr[i]*arr[j])+arr[k]);
  20. }
  21.  
  22. sort(lhs.begin(),lhs.end());
  23. sort(rhs.begin(),rhs.end());
  24. int sum=0;
  25. vector<int> :: iterator it1,it2;
  26. for(i=0;i<lhs.size();i++){
  27. sum+=(upper_bound(rhs.begin(),rhs.end(),lhs[i])-lower_bound(rhs.begin(),rhs.end(),lhs[i]));
  28. }
  29. cout<<sum<<endl;
  30. return 0;
  31. }
  32.  
Time limit exceeded #stdin #stdout 5s 4400KB
stdin
Standard input is empty
stdout
Standard output is empty