#include <iostream>
#include <vector>
using namespace std;
int main()
{
int n, m;
cin >> n >> m;
vector<bool> bulbs;
vector<pair<int, int>> switches;
for (int i = 0; i < n; i++) {
int input;
cin >> input;
bulbs.push_back(input);
}
for (int i = 0; i < m; i++) {
pair<int, int> pair;
cin >> pair.first >> pair.second;
switches.push_back(pair);
}
int result = 1;
for (int i = 0; i < m; i++) {
vector<bool> tempBulbs = bulbs;
if (tempBulbs[switches[i].first - 1] == tempBulbs[switches[i].second - 1]) {
result++;
tempBulbs[switches[i].first - 1] = !tempBulbs[switches[i].first - 1];
tempBulbs[switches[i].second - 1] = !tempBulbs[switches[i].second - 1];
}
for (int j = i + 1; j < m; j++) {
if (tempBulbs[switches[j].first - 1] == tempBulbs[switches[j].second - 1]) {
result++;
}
}
}
cout << result;
}
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 | #include <iostream> #include <vector> using namespace std; int main() { int n, m; cin >> n >> m; vector<bool> bulbs; vector<pair<int, int>> switches; for (int i = 0; i < n; i++) { int input; cin >> input; bulbs.push_back(input); } for (int i = 0; i < m; i++) { pair<int, int> pair; cin >> pair.first >> pair.second; switches.push_back(pair); } int result = 1; for (int i = 0; i < m; i++) { vector<bool> tempBulbs = bulbs; if (tempBulbs[switches[i].first - 1] == tempBulbs[switches[i].second - 1]) { result++; tempBulbs[switches[i].first - 1] = !tempBulbs[switches[i].first - 1]; tempBulbs[switches[i].second - 1] = !tempBulbs[switches[i].second - 1]; } for (int j = i + 1; j < m; j++) { if (tempBulbs[switches[j].first - 1] == tempBulbs[switches[j].second - 1]) { result++; } } } cout << result; } |
English