Submission #50578780

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0), cout.tie(0);

  int n;
  cin >> n;
  map<vector<int>, int> cnt;
  long long ans = 0;
  int zero = 0;
  for (int i = 0; i < n; ++i) {
    vector<int> p;
    int a;
    cin >> a;
    if (a == 0) {
      ans += i;
      ++zero;
      continue;
    }
    for (int j = 2; j * j <= a; ++j) {
      int q = 0;
      while (a % j == 0) {
        q ^= 1;
        a /= j;
      }
      if (q) p.push_back(j);
    }
    if (a > 1) p.push_back(a);
    ans += (cnt[p]++) + zero;
  }
  cout << ans << '\n';

  return 0;
}

Submission Info

Submission Time
Task D - Square Pair
User achvanov
Language C++ 20 (gcc 12.2)
Score 400
Code Size 647 Byte
Status AC
Exec Time 155 ms
Memory 16796 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 2
AC × 17
Set Name Test Cases
Sample 00_sample_01.txt, 00_sample_02.txt
All 00_sample_01.txt, 00_sample_02.txt, 01_test_01.txt, 01_test_02.txt, 01_test_03.txt, 01_test_04.txt, 01_test_05.txt, 01_test_06.txt, 01_test_07.txt, 01_test_08.txt, 01_test_09.txt, 01_test_10.txt, 01_test_11.txt, 01_test_12.txt, 01_test_13.txt, 01_test_14.txt, 01_test_15.txt
Case Name Status Exec Time Memory
00_sample_01.txt AC 1 ms 3560 KiB
00_sample_02.txt AC 1 ms 3476 KiB
01_test_01.txt AC 140 ms 16796 KiB
01_test_02.txt AC 23 ms 3500 KiB
01_test_03.txt AC 6 ms 3480 KiB
01_test_04.txt AC 75 ms 9740 KiB
01_test_05.txt AC 96 ms 10684 KiB
01_test_06.txt AC 119 ms 11524 KiB
01_test_07.txt AC 26 ms 3424 KiB
01_test_08.txt AC 26 ms 3424 KiB
01_test_09.txt AC 1 ms 3548 KiB
01_test_10.txt AC 1 ms 3492 KiB
01_test_11.txt AC 7 ms 4464 KiB
01_test_12.txt AC 155 ms 12888 KiB
01_test_13.txt AC 100 ms 11008 KiB
01_test_14.txt AC 11 ms 4960 KiB
01_test_15.txt AC 120 ms 11840 KiB