blob: 22d8c1543707c2ff43c630158bdf036adb44475c (
plain)
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
|
#include <algorithm>
#include <cstdint>
#include <iostream>
#include <map>
#include <queue>
#include <ranges>
#include <set>
#include <sstream>
#include <string>
#include <string_view>
#include <vector>
using namespace std;
class KeyLock {
public:
KeyLock() { fill(a, a+5, 0); }
void addrow(const string& s) {
for (int i = 0; i < 5; i++)
a[i] += s[i] == '#';
}
bool operator&(const KeyLock& other) const {
for (int i = 0; i < 5; i++)
if (a[i] + other.a[i] > 5)
return false;
return true;
}
private:
int a[5];
};
int main() {
vector<KeyLock> keys, locks;
string line;
while (getline(cin, line)) {
KeyLock k;
bool key = line[0] == '.';
for (int i = 0; i < 5; i++) {
getline(cin, line);
k.addrow(line);
}
if (key) keys.push_back(k);
else locks.push_back(k);
getline(cin, line);
getline(cin, line);
}
int tot = 0;
for (auto k : keys)
for (auto l : locks)
tot += (k & l) ? 1 : 0;
cout << tot << endl;
return 0;
}
|
