#include<bits/stdc++.h> #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 PB push_back #define EB emplace_back #define FI first #define SE second #define umap unordered_map #define uset unordered_set #define vi vector<int> #define vvi vector<vi> #define vll vector<ll> #define vvll vector<vll> #define vpii vector<pii> #define pii pair<int, int> #define pll pair<ll, ll> #define ALL(X) (X).begin(),(X).end() #ifndef DEBUG #define endl (char)10 #endif using namespace std; using ll = long long; using ld = long double; template <class T> istream& operator>> (istream& is, vector<T>& vec){ FOR(i,0,vec.size()) is >> vec[i]; return is; } template <class T> ostream& operator<< (ostream& os, vector<T>& vec){ for(auto& t : vec) os << t << " "; return os; } template<class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p){ os << p.FI << " " << p.SE; return os; } template<class T, class U> istream& operator>> (istream& is, pair<T, U>& p){ is >> p.FI >> p.SE; return is; } int main () { ios_base::sync_with_stdio(false); cin.tie(0); int n, k; cin >> n >> k; vi V(n); cin >> V; set<vi> B; B.insert(V); queue<vi> Q; Q.push(V); vpii S(k); FOR(i,0,k) {cin >> S[i].FI >> S[i].SE; S[i].FI--; S[i].SE--; } while(!Q.empty()){ auto c = Q.front(); Q.pop(); for(auto& s : S){ if (c[s.FI] == c[s.SE]){ c[s.FI] = c[s.SE] = 1 - c[s.SE]; if (B.count(c) == 0){ B.insert(c); Q.push(c); } c[s.FI] = c[s.SE] = 1 - c[s.SE]; } } } cout << B.size() << endl; }
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 | #include<bits/stdc++.h> #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 PB push_back #define EB emplace_back #define FI first #define SE second #define umap unordered_map #define uset unordered_set #define vi vector<int> #define vvi vector<vi> #define vll vector<ll> #define vvll vector<vll> #define vpii vector<pii> #define pii pair<int, int> #define pll pair<ll, ll> #define ALL(X) (X).begin(),(X).end() #ifndef DEBUG #define endl (char)10 #endif using namespace std; using ll = long long; using ld = long double; template <class T> istream& operator>> (istream& is, vector<T>& vec){ FOR(i,0,vec.size()) is >> vec[i]; return is; } template <class T> ostream& operator<< (ostream& os, vector<T>& vec){ for(auto& t : vec) os << t << " "; return os; } template<class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p){ os << p.FI << " " << p.SE; return os; } template<class T, class U> istream& operator>> (istream& is, pair<T, U>& p){ is >> p.FI >> p.SE; return is; } int main () { ios_base::sync_with_stdio(false); cin.tie(0); int n, k; cin >> n >> k; vi V(n); cin >> V; set<vi> B; B.insert(V); queue<vi> Q; Q.push(V); vpii S(k); FOR(i,0,k) {cin >> S[i].FI >> S[i].SE; S[i].FI--; S[i].SE--; } while(!Q.empty()){ auto c = Q.front(); Q.pop(); for(auto& s : S){ if (c[s.FI] == c[s.SE]){ c[s.FI] = c[s.SE] = 1 - c[s.SE]; if (B.count(c) == 0){ B.insert(c); Q.push(c); } c[s.FI] = c[s.SE] = 1 - c[s.SE]; } } } cout << B.size() << endl; } |