From adb97626f5553013e2758aeb136c8ffd28436627 Mon Sep 17 00:00:00 2001 From: omagdy7 Date: Mon, 18 Jul 2022 22:11:40 +0200 Subject: Solved a couple of problems --- codechef/sticks/main.cpp | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 codechef/sticks/main.cpp (limited to 'codechef/sticks/main.cpp') diff --git a/codechef/sticks/main.cpp b/codechef/sticks/main.cpp new file mode 100644 index 0000000..b5fae0e --- /dev/null +++ b/codechef/sticks/main.cpp @@ -0,0 +1,42 @@ +#include + +using namespace std; + +int main() { + ios_base::sync_with_stdio(false); + cin.tie(NULL); + int tt; + cin >> tt; + while (tt--) { + int n; + cin >> n; + vector v(n); + map mp; + for (int &x : v) { + cin >> x; + mp[x]++; + } + vector pair; + set vis; + int sq = 0; + sort(v.begin(), v.end()); + for(int i = v.size() - 1; i > 0; i--) { + int elem = v[i]; + if(mp[elem] >= 4) { + sq = elem; + } + if(mp[elem] >= 2 && !vis.count(elem)) { + if(pair.size() == 2) { + break; + } + pair.push_back(elem); + vis.insert(elem); + } + } + if(pair.size() == 2) { + cout << max(pair[0] * pair[1], sq * sq) << endl; + } else { + cout << -1 << endl; + } + } +} -- cgit v1.2.3