#include "museum.h"
#include<map>
#include<vector>
#include<iostream>
#define ll long long
#define rep(i,a,b) for(int i = a;i < b;i++)
#define MAXN 300003
using namespace std;
ll ar[MAXN][3];
map < ll,ll > freq;
long long CountSimilarPairs(std::vector<int> B, std::vector<int> T, std::vector<int> G) {
int n = B.size();
ll ans = 0;
rep(i,0,n)
{
ar[i][0] = B[i];
ar[i][1] = T[i];
ar[i][2] = G[i];
}
// 1 stixio kino
rep(t,0,3)
{
freq.clear();
rep(i,0,n)
{
freq[ar[i][t]]++;
}
for(map < ll,ll >::iterator it = freq.begin();it != freq.end();it++)
{
ll val = it->second;
ans += (val*(val-1))/2;
}
}
// cout << ans << endl;
// 2 stixia kina
freq.clear();
rep(i,0,n)
{
freq[ar[i][0]*101 + ar[i][1]]++;
}
for(map < ll,ll >::iterator it = freq.begin();it != freq.end();it++)
{
ll val = it->second;
ans -= (val*(val-1))/2;
}
freq.clear();
rep(i,0,n)
{
freq[ar[i][0]*101 + ar[i][2]]++;
}
for(map < ll,ll >::iterator it = freq.begin();it != freq.end();it++)
{
ll val = it->second;
ans -= (val*(val-1))/2;
}
// cout << ans << endl;
freq.clear();
rep(i,0,n)
{
freq[ar[i][1]*101 + ar[i][2]]++;
}
for(map < ll,ll >::iterator it = freq.begin();it != freq.end();it++)
{
ll val = it->second;
ans -= (val*(val-1))/2;
}
// 3 stixia kina
freq.clear();
rep(i,0,n)
{
freq[ar[i][0]*12000+ar[i][1]*101+ar[i][2]]++;
}
for(map < ll,ll >::iterator it = freq.begin();it != freq.end();it++)
{
ll val = it->second;
ans += (val*(val-1))/2;
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
512 KB |
Output is correct |
2 |
Correct |
6 ms |
384 KB |
Output is correct |
3 |
Correct |
6 ms |
472 KB |
Output is correct |
4 |
Correct |
6 ms |
384 KB |
Output is correct |
5 |
Correct |
6 ms |
384 KB |
Output is correct |
6 |
Correct |
13 ms |
512 KB |
Output is correct |
7 |
Correct |
8 ms |
512 KB |
Output is correct |
8 |
Correct |
7 ms |
512 KB |
Output is correct |
9 |
Correct |
19 ms |
640 KB |
Output is correct |
10 |
Correct |
9 ms |
640 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
512 KB |
Output is correct |
2 |
Correct |
6 ms |
384 KB |
Output is correct |
3 |
Correct |
6 ms |
472 KB |
Output is correct |
4 |
Correct |
6 ms |
384 KB |
Output is correct |
5 |
Correct |
6 ms |
384 KB |
Output is correct |
6 |
Correct |
13 ms |
512 KB |
Output is correct |
7 |
Correct |
8 ms |
512 KB |
Output is correct |
8 |
Correct |
7 ms |
512 KB |
Output is correct |
9 |
Correct |
19 ms |
640 KB |
Output is correct |
10 |
Correct |
9 ms |
640 KB |
Output is correct |
11 |
Correct |
17 ms |
1024 KB |
Output is correct |
12 |
Correct |
20 ms |
1792 KB |
Output is correct |
13 |
Correct |
88 ms |
6392 KB |
Output is correct |
14 |
Correct |
153 ms |
9096 KB |
Output is correct |
15 |
Correct |
250 ms |
13424 KB |
Output is correct |
16 |
Correct |
77 ms |
9964 KB |
Output is correct |
17 |
Correct |
144 ms |
9964 KB |
Output is correct |
18 |
Correct |
332 ms |
16004 KB |
Output is correct |
19 |
Correct |
455 ms |
21128 KB |
Output is correct |
20 |
Correct |
413 ms |
21220 KB |
Output is correct |