fork download
  1. #include "bits/stdc++.h"
  2. using namespace std;
  3. int main(){
  4. cin.tie(NULL);
  5. ios_base::sync_with_stdio(0);
  6. int n; cin >> n;
  7. vector<int> nums(n);
  8. for (int &i : nums) cin >> i;
  9. int mx = *max_element(nums.begin(), nums.end());
  10. vector<int> cnt(mx * 2 + 1);
  11. for (int i : nums) cnt[i]++;
  12. int ans = 0;
  13. for (int i = 0; i * i <= mx * 2; ++i) {
  14. for (int j = 0; j < n; ++j) {
  15. cnt[nums[j]]--;
  16. if (nums[j] <= i * i) {
  17. ans += cnt[i * i - nums[j]];
  18. }
  19. cnt[nums[j]]++;
  20. }
  21. }
  22. cout << ans / 2 << "\n";
  23. }
Success #stdin #stdout 0.01s 5308KB
stdin
5
1 2 3 4 5
stdout
2