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

Задача F. Үштіктер

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

1 second

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

256 megabytes

Сізге ұзындығы $n$ болатын, бүтін сандардан тұратын $a$ массиві берілген. Массивтегі барлық сандар әртүрлі. Сізге $i \ne j$, $i \ne k$, $j \ne k$ және $a_i*a_j=a_k^2$ орындалатын $i$, $j$, $k$ үштіктер санын табу керек.

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

Бірінші жолда $n$ $(3 <= n <= 10^5)$ бүтін саны берілген. Келесі жолда $a_1$, $a_2$, $\dots$, $a_n$ $(1 <= a_i <= 3 * 10^5)$ болатын $a$ массиві берілген.

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

Есептің жауабын шығарыңыз.

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

Бұл есеп $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();
   }
}