#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <queue>
using namespace std;
int main(){
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int n, m;
cin >> n >> m;
vector <bool> zar(n, false);
for (int i = 0; i < n; i++){
int x;
cin >> x;
if (x == 1){
zar[i] = true;
}
}
vector <pair <int, int>> przelaczniki (m);
for (int i = 0; i < m; i++){
cin >> przelaczniki[i].first >> przelaczniki[i].second;
przelaczniki[i].first--; przelaczniki[i].second--;
}
map <vector<bool>, int> odwiedzone;
odwiedzone[zar] = 1;
queue <vector<bool>> q;
q.push(zar);
while (!q.empty()){
vector <bool> aktualny = q.front();
q.pop();
for (int i = 0; i < m; i++){
if (aktualny[przelaczniki[i].first] == aktualny[przelaczniki[i].second]){
aktualny[przelaczniki[i].first] = !aktualny[przelaczniki[i].first];
aktualny[przelaczniki[i].second] = !aktualny[przelaczniki[i].second];
if (odwiedzone[aktualny] == 0){
odwiedzone[aktualny] = 1;
q.push(aktualny);
}
aktualny[przelaczniki[i].first] = !aktualny[przelaczniki[i].first];
aktualny[przelaczniki[i].second] = !aktualny[przelaczniki[i].second];
}
}
}
cout << odwiedzone.size()%(1000000000+7) << "\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 | #include <iostream> #include <vector> #include <algorithm> #include <map> #include <queue> using namespace std; int main(){ ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); int n, m; cin >> n >> m; vector <bool> zar(n, false); for (int i = 0; i < n; i++){ int x; cin >> x; if (x == 1){ zar[i] = true; } } vector <pair <int, int>> przelaczniki (m); for (int i = 0; i < m; i++){ cin >> przelaczniki[i].first >> przelaczniki[i].second; przelaczniki[i].first--; przelaczniki[i].second--; } map <vector<bool>, int> odwiedzone; odwiedzone[zar] = 1; queue <vector<bool>> q; q.push(zar); while (!q.empty()){ vector <bool> aktualny = q.front(); q.pop(); for (int i = 0; i < m; i++){ if (aktualny[przelaczniki[i].first] == aktualny[przelaczniki[i].second]){ aktualny[przelaczniki[i].first] = !aktualny[przelaczniki[i].first]; aktualny[przelaczniki[i].second] = !aktualny[przelaczniki[i].second]; if (odwiedzone[aktualny] == 0){ odwiedzone[aktualny] = 1; q.push(aktualny); } aktualny[przelaczniki[i].first] = !aktualny[przelaczniki[i].first]; aktualny[przelaczniki[i].second] = !aktualny[przelaczniki[i].second]; } } } cout << odwiedzone.size()%(1000000000+7) << "\n"; } |
English