-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCustom_Hash_Unordered_map_for_pairs.cpp
More file actions
173 lines (150 loc) · 5.03 KB
/
Custom_Hash_Unordered_map_for_pairs.cpp
File metadata and controls
173 lines (150 loc) · 5.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
// #pragma GCC optimize("Ofast")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,fma")
// #pragma GCC optimize("unroll-loops")
#include "bits/stdc++.h"
using namespace std;
// #include <ext/pb_ds/assoc_container.hpp>
// #include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
string to_string(char c) {
return string(1, c);
}
template <typename A, typename B>
string to_string(pair<A, B> p) {
return "(" + to_string(p.first) + ", " + to_string(p.second) + ")";
}
template <size_t N>
string to_string(bitset<N> v) {
string res = "";
for (size_t i = 0; i < N; i++) {
res += static_cast<char>('0' + v[i]);
}
return res;
}
template <typename A>
string to_string(A v) {
bool first = true;
string res = "{";
for (const auto &x : v) {
if (!first) {
res += ", ";
}
first = false;
res += to_string(x);
}
res += "}";
return res;
}
#define rep(i, a, b) for(int i = (a); i < (b); ++i)
#define per(i, a, b) for(int i = (a) ; i >= (b); i--)
#define forn(i,n) rep(i,0,(n))
#define rof(i,n) per(i,(n)-1,0)
#define dbg(x) cout << #x << "=" << x << endl
#define dbg2(x, y) cout << #x << "=" << x << "," << #y << "=" << y << endl
#define dbg3(x, y,z) cout << #x << "=" << x << "," << #y << "=" << y <<"," << #z << "=" << z << endl
#define ff first
#define ss second
#define mp make_pair
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
#define endl "\n"
#define int long long
#define ll long long
#define pii pair<int,int>
#define setbits(x) __builtin_popcountll(x)
#define pqb priority_queue<int> // maxheap
#define pqs priority_queue<int,vector<int>,greater<int>> // minheap
#define piipqs priority_queue<pii,vector<pii>,greater<pii>> // minheap for pair<int,int>
#define piipqb priority_queue<pii> // maxheap for pair<int,int>
#define mod1 1000000007
#define mod2 998244353
#define memt(a) memset(a,true,sizeof(a))
#define memf(a) memset(a,false,sizeof(a))
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,0x7f,sizeof(a))
#define precise(x,y) fixed<<setprecision(y)<<x
#define FIO ios_base::sync_with_stdio(0); cin.tie(0)
#define getunique(v) {sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end());}
#define NO {cout<<"NO"<<endl; return;}
#define YES {cout<<"YES"<<endl; return;}
#define NEG1 {cout<<"-1"<<endl; return;}
// #define oset tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update> // set
// #define osetpii tree<pii, null_type,less<pii>, rb_tree_tag,tree_order_statistics_node_update> //like multiset
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
// for long long use mt19937_64 and usage - just do rng()
typedef tuple<int, int> tpl;
const ll inf = 2e18; //2e18
const double epsilon = 1e-7;
template<typename T, typename T1>T amax(T &a, T1 b) {if (b > a)a = b; return a;}
template<typename T, typename T1>T amin(T &a, T1 b) {if (b < a)a = b; return a;}
inline ll gcd(ll a, ll b) {return (b == 0) ? a : gcd(b, a % b);}
int dx[] = { -1, 0, 1, 0};
int dy[] = {0, 1, 0, -1};
// **************************** Code Begins **************************** //
// 1 => 2 3 4 => 1 2 3 4
// 2 => 1 3 4 => 1 2 3 4
// 3 => 2 4 => 2 3 4
// 4 => 2 3 => 2 3 4
// 5 => 1 2
// 6 => 1 2
// https://codeforces.com/problemset/problem/154/C
struct pair_hash
{
template <class T1, class T2>
size_t operator() (const pair<T1, T2> &pair) const {
return std::hash<T1>()(pair.first) ^ std::hash<T2>()(pair.second);
}
};
const int p1 = 31, p2 = 43;
int calc_hash(vector<int> &adj, int p[], int mod)
{
int hash = 0;
for (int nbr : adj)
hash = (hash + p[nbr] * nbr) % mod;
return hash;
}
void solve()
{
int n, m;
cin >> n >> m;
int p_pow1[n + 1] = {1};
int p_pow2[n + 1] = {1};
rep(i, 1, n + 1)
{
p_pow1[i] = p_pow1[i - 1] * p1 % mod1;
p_pow2[i] = p_pow2[i - 1] * p2 % mod2;
}
vector<int> adj[n + 1];
forn(i, m)
{
int x, y;
cin >> x >> y;
adj[x].push_back(y);
adj[y].push_back(x);
}
unordered_map<pii, int, pair_hash> cnt;
cnt.reserve(4 * n);
rep(i, 1, n + 1)
{
int wfhash1 = calc_hash(adj[i], p_pow1, mod1);
int wfhash2 = calc_hash(adj[i], p_pow2, mod2);
cnt[ {wfhash1, wfhash2}]++;
adj[i].push_back(i);
int fhash1 = calc_hash(adj[i], p_pow1, mod1);
int fhash2 = calc_hash(adj[i], p_pow2, mod2);
cnt[ {fhash1, fhash2}]++;
}
int ans = 0;
for (auto p : cnt)
ans += (p.ss * (p.ss - 1)) / 2;
cout << ans << endl;
}
signed main()
{
FIO;
int tt = 1;
// cin >> tt;
for (int i = 1; i <= tt; i++)
solve();
}