// Author : Jakub Rożek // Task : Żarówki // Contest : PA 2024 r5 C // Memory : O(n) // Time : O(n ) // Solution : Rozwiązanie dobre #include "bits/stdc++.h" using namespace std; using LL = long long; template <typename T> using P = pair<T, T>; template <typename T> using VV = vector<vector<T>>; #define all(x) x.begin(), x.end() #define FOR(i,a,b) for(int i=(a); i<=(b); ++i) #define FORD(i,a,b) for(int i=(a); i>=(b); --i) #define REP(i,n) for(int i=0; i<(n); ++i) #define ssize(x) int((x).size()) #ifdef DEBUG template <typename T1, typename T2> auto&operator<<(auto&o,pair<T1,T2>p){return o<<'('<<p.first<<", "<<p.second<<")";} auto operator<<(auto&o,auto x)->decltype(x.end(),o){o<<"{";for(auto e:x)o<<","<<e;return o<<"}";} #define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(x) #else #define debug(...) {} #endif // const int INF = 1'000'000'009; const LL mod = 1'000'000'007; const int N = 200'000; int n, m, x, y; LL k[2][2]; LL odp, odpp, r; bool kolor_grafu[N+1]; bool kolor[N+1]; bool dwudzielny; int odw[N+1]; LL silnia[N+1]; LL odwsilnia[N+1]; VV<int> graf; LL pot(LL a, LL b) { LL w = 1; while (b) { if (b % 2) w = (w * a) % mod; a = (a * a) % mod; b /= 2; } return w; } LL n_po_k(LL a, LL b) { LL w = 1; w *= silnia[a]; w %= mod; w *= odwsilnia[b]; w %= mod; w *= odwsilnia[a-b]; w %= mod; return w; } void zlicz_kolory(int a, int p) { if (odw[a]) { if (kolor_grafu[a] != p) dwudzielny = false; return; } odw[a] = true; kolor_grafu[a] = p; k[p][kolor[a]]++; for (auto i:graf[a]) { zlicz_kolory(i, 1-p); } } void solution() { cin >> n >> m; graf.resize(n+1); FOR (i, 1, n) cin >> kolor[i]; REP (i, m) { cin >> x >> y; graf[x].push_back(y); graf[y].push_back(x); } silnia[0] = 1; FOR (i, 1, n) silnia[i] = (silnia[i-1] * i) % mod; odwsilnia[n] = pot(silnia[n], mod-2); FORD (i, n-1, 0) odwsilnia[i] = (odwsilnia[i+1] * (i+1)) % mod; odp = 1; FOR (i, 1, n) { if (odw[i]) continue; k[0][0] = 0; k[0][1] = 0; k[1][0] = 0; k[1][1] = 0; dwudzielny = true; zlicz_kolory(i, 0); if (dwudzielny) { r = k[0][1] - k[1][1]; odpp = 0; FOR (j, 0, k[0][0]+k[0][1]) { LL jj = j - r; if (jj < 0 || jj > k[1][0]+k[1][1]) continue; odpp += n_po_k(k[0][0]+k[0][1], j) * n_po_k(k[1][0]+k[1][1], jj) % mod; odpp %= mod; } odp *= odpp; odp %= mod; } else { odp *= pot(2, k[0][0]+k[0][1]+k[1][0]+k[1][1]-1); odp %= mod; } } cout << odp; return; } int main() { cin.tie(0)->sync_with_stdio(0); int tests = 1; // cin>>tests; FOR (i, 1, tests) { solution(); } return 0; }
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 | // Author : Jakub Rożek // Task : Żarówki // Contest : PA 2024 r5 C // Memory : O(n) // Time : O(n ) // Solution : Rozwiązanie dobre #include "bits/stdc++.h" using namespace std; using LL = long long; template <typename T> using P = pair<T, T>; template <typename T> using VV = vector<vector<T>>; #define all(x) x.begin(), x.end() #define FOR(i,a,b) for(int i=(a); i<=(b); ++i) #define FORD(i,a,b) for(int i=(a); i>=(b); --i) #define REP(i,n) for(int i=0; i<(n); ++i) #define ssize(x) int((x).size()) #ifdef DEBUG template <typename T1, typename T2> auto&operator<<(auto&o,pair<T1,T2>p){return o<<'('<<p.first<<", "<<p.second<<")";} auto operator<<(auto&o,auto x)->decltype(x.end(),o){o<<"{";for(auto e:x)o<<","<<e;return o<<"}";} #define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(x) #else #define debug(...) {} #endif // const int INF = 1'000'000'009; const LL mod = 1'000'000'007; const int N = 200'000; int n, m, x, y; LL k[2][2]; LL odp, odpp, r; bool kolor_grafu[N+1]; bool kolor[N+1]; bool dwudzielny; int odw[N+1]; LL silnia[N+1]; LL odwsilnia[N+1]; VV<int> graf; LL pot(LL a, LL b) { LL w = 1; while (b) { if (b % 2) w = (w * a) % mod; a = (a * a) % mod; b /= 2; } return w; } LL n_po_k(LL a, LL b) { LL w = 1; w *= silnia[a]; w %= mod; w *= odwsilnia[b]; w %= mod; w *= odwsilnia[a-b]; w %= mod; return w; } void zlicz_kolory(int a, int p) { if (odw[a]) { if (kolor_grafu[a] != p) dwudzielny = false; return; } odw[a] = true; kolor_grafu[a] = p; k[p][kolor[a]]++; for (auto i:graf[a]) { zlicz_kolory(i, 1-p); } } void solution() { cin >> n >> m; graf.resize(n+1); FOR (i, 1, n) cin >> kolor[i]; REP (i, m) { cin >> x >> y; graf[x].push_back(y); graf[y].push_back(x); } silnia[0] = 1; FOR (i, 1, n) silnia[i] = (silnia[i-1] * i) % mod; odwsilnia[n] = pot(silnia[n], mod-2); FORD (i, n-1, 0) odwsilnia[i] = (odwsilnia[i+1] * (i+1)) % mod; odp = 1; FOR (i, 1, n) { if (odw[i]) continue; k[0][0] = 0; k[0][1] = 0; k[1][0] = 0; k[1][1] = 0; dwudzielny = true; zlicz_kolory(i, 0); if (dwudzielny) { r = k[0][1] - k[1][1]; odpp = 0; FOR (j, 0, k[0][0]+k[0][1]) { LL jj = j - r; if (jj < 0 || jj > k[1][0]+k[1][1]) continue; odpp += n_po_k(k[0][0]+k[0][1], j) * n_po_k(k[1][0]+k[1][1], jj) % mod; odpp %= mod; } odp *= odpp; odp %= mod; } else { odp *= pot(2, k[0][0]+k[0][1]+k[1][0]+k[1][1]-1); odp %= mod; } } cout << odp; return; } int main() { cin.tie(0)->sync_with_stdio(0); int tests = 1; // cin>>tests; FOR (i, 1, tests) { solution(); } return 0; } |