Районная олимпиада 2019-2020 информатика

Задача F. Тройки

Ограничение по времени:

1 секунда

Ограничение по памяти:

256 мегабайт

Дан массив $a$ длины $n$, который состоит из целых натуральных чисел. Все числа в массиве различные. Вам нужно посчитать количество троек $i$, $j$, $k$, что $i \ne j$, $i \ne k$, $j \ne k$ и $a_i*a_j=a_k^2$.

Формат входного файла

В первой строке задано одно целое число $n$ $(3 <= n <= 10^5)$. В следующей строке задано $n$ целых натуральных чисел $a_1$, $a_2$, $\dots$, $a_n$ $(1 <= a_i <= 3 * 10^5)$.

Формат выходного файла

Выведите одно целое число — ответ на задачу.

Система оценки

Данная задача содержит $10$ тестов. Каждый тест оценивается в $10$ баллов:

1. $n <= 100$, $1 <= a_i <= 500$. Тесты с номерами 1-3.
2. $n <= 5000$, $1 <= a_i <= 10000$. Тесты с номерами 4-6.
3. $n <= 10^5$, $1 <= a_i <= 300000$. Тесты с номерами 7-10.

Пример:

Вход

6
6 3 9 4 12 27

Ответ

6

( Aibar Kuanyshbay )

Комментарий/решение

Здравствуйте, эту задачу возможно решить за такие ограничение?

Решение на 60 баллов за O(n^2)

показать/скрыть код

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define pf push_front
#define F first
#define S second
#define ll long long
#define all(x) x.begin(), x.end() 
#define pii pair<int, int>
#define unm unordered_map
#define pll pair<ll, ll>
#define speed ios_base::sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL)
#define rand mt19937 rng(chrono::steady_clock::now().time_since_epoch().count())
#define file freopen("a.txt", "r", stdin), freopen("A.txt", "w", stdout);
const ll SIZE = (ll)3 * 1e5 + 5;
void solve(){
    int n, c = 0;
    cin >> n;
    int a[n];
    int count[SIZE]{0};
    for(int i = 0; i < n; i++){
    	cin >> a[i];
    	count[a[i]] = 1;
    }
    for(int i = 0; i < n; i++){
    	for(int j = 0; j < n; j++){
    		ll check = sqrt(a[i] * a[j]);
    		if(i == j) continue;
    		if(check * check != a[i] * a[j]) continue;
    		if(count[check] == 1){
    			c++;
    		}
    	}
    }
    cout << c;
} 
int main(){
   speed;
   int t = 1;
   while(t--){
      solve();
   }
}