#include<bits/stdc++.h> #define MAX 3001 #define MOD 1000000007 using namespace std; int data[MAX][MAX]; int dostepne[MAX][MAX]; vector<pair<int,int>>pary; long long wynik; int n, k; void init(); bool czterykierunki(int a, int b); int main() { ios_base::sync_with_stdio(0); string s; int i, j; cin >> n >> k; init(); for(i = 0; i <= n; i++) { getline(cin, s); for(j = 1; j <= s.length(); j++) { if(s[j-1] == '#') { data[i][j] = 1; } else { data[i][j] = 0; } } } if(k==1) { wynik = 0; for(int i = 1; i <= n; i++) for(int j = 1; j <= n; ++j) { if(data[i][j] == 0 && czterykierunki(i, j)) { wynik++; } } } else if(k == 2) { long long pola = 0; for(int i = 1; i <= n; i++) { for(int j = 1; j <= n; j++) { if(data[i][j] == 0 && czterykierunki(i, j)) { dostepne[i][j] = 0; pola += 1; pary.push_back(make_pair(i, j)); } else if(data[i][j] == 0) { dostepne[i][j] = 1; } } } pola -= 1; wynik = 0; for(int count = 0; count < pary.size(); count++) { wynik += pola; int i = pary[count].first; int j = pary[count].second; if(dostepne[i][j+1] || dostepne[i][j-1] || dostepne[i+1][j] || dostepne[i-1][j]) { wynik += 1; } pola -= 1; } } cout << wynik % MOD << endl; } void init(){ for(int i = 0; i < n; i++) { dostepne[0][i] = 0; data[0][i] = 0; } for(int i = 0; i < n; i++) { dostepne[n + 1][i] = 0; data[n + 1][i] = 0; } for(int i = 0; i < n; i++) { dostepne[i][0] = 0; data[i][0] = 0; } for(int i = 0; i < n; i++) { dostepne[i][n + 1] = 0; data[i][n + 1] = 0; } } bool czterykierunki(int a, int b) { return data[a][b+1]||data[a][b-1]||data[a + 1][b]||data[a-1][b]; }
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 | #include<bits/stdc++.h> #define MAX 3001 #define MOD 1000000007 using namespace std; int data[MAX][MAX]; int dostepne[MAX][MAX]; vector<pair<int,int>>pary; long long wynik; int n, k; void init(); bool czterykierunki(int a, int b); int main() { ios_base::sync_with_stdio(0); string s; int i, j; cin >> n >> k; init(); for(i = 0; i <= n; i++) { getline(cin, s); for(j = 1; j <= s.length(); j++) { if(s[j-1] == '#') { data[i][j] = 1; } else { data[i][j] = 0; } } } if(k==1) { wynik = 0; for(int i = 1; i <= n; i++) for(int j = 1; j <= n; ++j) { if(data[i][j] == 0 && czterykierunki(i, j)) { wynik++; } } } else if(k == 2) { long long pola = 0; for(int i = 1; i <= n; i++) { for(int j = 1; j <= n; j++) { if(data[i][j] == 0 && czterykierunki(i, j)) { dostepne[i][j] = 0; pola += 1; pary.push_back(make_pair(i, j)); } else if(data[i][j] == 0) { dostepne[i][j] = 1; } } } pola -= 1; wynik = 0; for(int count = 0; count < pary.size(); count++) { wynik += pola; int i = pary[count].first; int j = pary[count].second; if(dostepne[i][j+1] || dostepne[i][j-1] || dostepne[i+1][j] || dostepne[i-1][j]) { wynik += 1; } pola -= 1; } } cout << wynik % MOD << endl; } void init(){ for(int i = 0; i < n; i++) { dostepne[0][i] = 0; data[0][i] = 0; } for(int i = 0; i < n; i++) { dostepne[n + 1][i] = 0; data[n + 1][i] = 0; } for(int i = 0; i < n; i++) { dostepne[i][0] = 0; data[i][0] = 0; } for(int i = 0; i < n; i++) { dostepne[i][n + 1] = 0; data[i][n + 1] = 0; } } bool czterykierunki(int a, int b) { return data[a][b+1]||data[a][b-1]||data[a + 1][b]||data[a-1][b]; } |