#include <bits/stdc++.h> using namespace std; #define st first #define nd second #define ll long long const ll maxn = 2e5 + 7, MOD = 1e9 + 7; ll n, m, a, b, res, wyn, roz; vector<int> G[maxn]; ll t[maxn], dwojka[maxn], licze[maxn][3], jeden[maxn], liczzap[maxn][3], dwumian[3][maxn]; ll vis[maxn]; bool dwu[maxn]; void DFS(int w, int dist, int pocz){ licze[pocz][dist % 2]++; if(t[w] == 1) liczzap[pocz][dist % 2]++; vis[w] = dist % 2 + 1; for(int i = 0; i < G[w].size(); i++){ if(!vis[G[w][i]]) DFS(G[w][i], dist + 1, pocz); else if(vis[G[w][i]] == vis[w]) dwu[pocz] = 1; } } int gcdExtended(int a, int b, int &x, int &y) { if (a == 0) { x = 0; y = 1; return b; } int x1, y1; int gcd = gcdExtended(b % a, a, x1, y1); x = y1 - (b / a) * x1; y = x1; return gcd; } int inwersja(int a, int m) { int x, y; int gcd = gcdExtended(a, m, x, y); int result = (x % m + m) % m; return result; } int podzielnprzezkmod(ll n, ll k){ ll inw_k = inwersja(k%MOD, MOD); return (((n % MOD) * inw_k)%MOD); } void policzdwumian(int a, ll n){ dwumian[a][0] = 1; for(int i = 1; i <= n; i++){ dwumian[a][i] = podzielnprzezkmod(dwumian[a][i - 1] * (n - i + 1), i); } } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n >> m; for(int i = 1; i <= n; i++) cin >> t[i]; for(int i = 1; i <= m; i++){ cin >> a >> b; G[a].push_back(b); G[b].push_back(a); } dwojka[0] = 1; for(int i = 1; i <= n; i++){ dwojka[i] = ((dwojka[i - 1] * 2) % MOD); } res = 1; for(int l = 1; l <= n; l++){ // cout << res << '\n'; if(!vis[l]){ DFS(l, 0, l); if(dwu[l]){ jeden[l] += res; res = (res * dwojka[licze[l][0] + licze[l][1] - 1]) % MOD; continue; // cout << l << " " <<dwojka[licze[l][0] + licze[l][1] - 1] << '\n'; } roz = liczzap[l][0] - liczzap[l][1]; if(roz >= 0){ policzdwumian(0, licze[l][0]); policzdwumian(1, licze[l][1]); for(int i = 0; i <= min(licze[l][1], licze[l][0] - roz); i++) wyn = (wyn + (dwumian[0][i + roz] * dwumian[1][i]) % MOD) % MOD; res = (res * wyn) % MOD; } else{ roz *= (-1); policzdwumian(0, licze[l][1]); policzdwumian(1, licze[l][0]); for(int i = 0; i <= min(licze[l][0], licze[l][1] - roz); i++) wyn = (wyn + (dwumian[0][i + roz] * dwumian[1][i]) % MOD) % MOD; res = (res * wyn) % MOD; } roz = 0; wyn = 0; } } cout << res << '\n'; }
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 | #include <bits/stdc++.h> using namespace std; #define st first #define nd second #define ll long long const ll maxn = 2e5 + 7, MOD = 1e9 + 7; ll n, m, a, b, res, wyn, roz; vector<int> G[maxn]; ll t[maxn], dwojka[maxn], licze[maxn][3], jeden[maxn], liczzap[maxn][3], dwumian[3][maxn]; ll vis[maxn]; bool dwu[maxn]; void DFS(int w, int dist, int pocz){ licze[pocz][dist % 2]++; if(t[w] == 1) liczzap[pocz][dist % 2]++; vis[w] = dist % 2 + 1; for(int i = 0; i < G[w].size(); i++){ if(!vis[G[w][i]]) DFS(G[w][i], dist + 1, pocz); else if(vis[G[w][i]] == vis[w]) dwu[pocz] = 1; } } int gcdExtended(int a, int b, int &x, int &y) { if (a == 0) { x = 0; y = 1; return b; } int x1, y1; int gcd = gcdExtended(b % a, a, x1, y1); x = y1 - (b / a) * x1; y = x1; return gcd; } int inwersja(int a, int m) { int x, y; int gcd = gcdExtended(a, m, x, y); int result = (x % m + m) % m; return result; } int podzielnprzezkmod(ll n, ll k){ ll inw_k = inwersja(k%MOD, MOD); return (((n % MOD) * inw_k)%MOD); } void policzdwumian(int a, ll n){ dwumian[a][0] = 1; for(int i = 1; i <= n; i++){ dwumian[a][i] = podzielnprzezkmod(dwumian[a][i - 1] * (n - i + 1), i); } } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n >> m; for(int i = 1; i <= n; i++) cin >> t[i]; for(int i = 1; i <= m; i++){ cin >> a >> b; G[a].push_back(b); G[b].push_back(a); } dwojka[0] = 1; for(int i = 1; i <= n; i++){ dwojka[i] = ((dwojka[i - 1] * 2) % MOD); } res = 1; for(int l = 1; l <= n; l++){ // cout << res << '\n'; if(!vis[l]){ DFS(l, 0, l); if(dwu[l]){ jeden[l] += res; res = (res * dwojka[licze[l][0] + licze[l][1] - 1]) % MOD; continue; // cout << l << " " <<dwojka[licze[l][0] + licze[l][1] - 1] << '\n'; } roz = liczzap[l][0] - liczzap[l][1]; if(roz >= 0){ policzdwumian(0, licze[l][0]); policzdwumian(1, licze[l][1]); for(int i = 0; i <= min(licze[l][1], licze[l][0] - roz); i++) wyn = (wyn + (dwumian[0][i + roz] * dwumian[1][i]) % MOD) % MOD; res = (res * wyn) % MOD; } else{ roz *= (-1); policzdwumian(0, licze[l][1]); policzdwumian(1, licze[l][0]); for(int i = 0; i <= min(licze[l][0], licze[l][1] - roz); i++) wyn = (wyn + (dwumian[0][i + roz] * dwumian[1][i]) % MOD) % MOD; res = (res * wyn) % MOD; } roz = 0; wyn = 0; } } cout << res << '\n'; } |