백준 1037 약수, c++

https://www.acmicpc.net/problem/1037

1037번: 약수

풀이:

prime의 square number과 그 외를 구분해주면 되는 문제입니다.

 

#include <iostream>
#include <vector>
#include <algorithm>
#define fastio() ios::sync_with_stdio(0),cin.tie(nullptr),cout.tie(nullptr);
using namespace std;
int main() {
    fastio();
    int n; cin >> n;
    vector<int> v(n);
    for (int i = 0; i < n; i++) cin >> v[i];
    if (n == 1) cout << v[0] * v[0]; // square number of prime
    else {
        cout << *min_element(v.begin(), v.end()) * *max_element(v.begin(), v.end());;
    }
}