#include <iostream> #include <vector> #include <algorithm> #include <cmath> #include <unordered_map> #include <set> #include <dzialka.h> #include <message.h> using namespace std; typedef long long LL; LL n, m; vector<vector<LL>> pre, v; void makepre() { for(int i = 1; i <= n; i++) for(int j = 1; j <= m; j++) pre[i][j] = pre[i - 1][j] + pre[i][j - 1] - pre[i - 1][j - 1] + v[i][j]; } LL solve(int a, int b) { LL cnt = 0; for(int i = 0; i < a; i++) { for(int j = 0; j < b; j++) { LL ile = pre[a][b] - pre[a - i - 1][b] - pre[a][b - j - 1] + pre[a - i - 1][b - j - 1]; //cout << a << " " << b << " " << i << " " << j << " " << ile << endl; if(ile == (LL)(i + 1) * (LL)(j + 1)) cnt++; else break; } } return cnt; } int main() { if(MyNodeId() != 0) return 0; n = GetFieldHeight(); m = GetFieldWidth(); //cin >> n >> m; v.resize(n + 7), pre.resize(n + 7); for(int i = 0; i <= n + 5; i++) v[i].resize(m + 7, 0), pre[i].resize(m + 7, 0); for(int i = 0; i < n; i++) { for(int j = 0; j < m; j++) { LL czy; czy = (LL)IsUsableCell(i, j); //cin >> czy; v[i + 1][j + 1] = czy; } } makepre(); LL res = 0; for(int i = 1; i <= n; i++) { for(int j = 1; j <= m; j++) { if(v[i][j] == 0) continue; LL x = solve(i, j); res += x; //cout << i << " " << j << " " << x << endl; } } printf("%lld", res); return 0; }
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 | #include <iostream> #include <vector> #include <algorithm> #include <cmath> #include <unordered_map> #include <set> #include <dzialka.h> #include <message.h> using namespace std; typedef long long LL; LL n, m; vector<vector<LL>> pre, v; void makepre() { for(int i = 1; i <= n; i++) for(int j = 1; j <= m; j++) pre[i][j] = pre[i - 1][j] + pre[i][j - 1] - pre[i - 1][j - 1] + v[i][j]; } LL solve(int a, int b) { LL cnt = 0; for(int i = 0; i < a; i++) { for(int j = 0; j < b; j++) { LL ile = pre[a][b] - pre[a - i - 1][b] - pre[a][b - j - 1] + pre[a - i - 1][b - j - 1]; //cout << a << " " << b << " " << i << " " << j << " " << ile << endl; if(ile == (LL)(i + 1) * (LL)(j + 1)) cnt++; else break; } } return cnt; } int main() { if(MyNodeId() != 0) return 0; n = GetFieldHeight(); m = GetFieldWidth(); //cin >> n >> m; v.resize(n + 7), pre.resize(n + 7); for(int i = 0; i <= n + 5; i++) v[i].resize(m + 7, 0), pre[i].resize(m + 7, 0); for(int i = 0; i < n; i++) { for(int j = 0; j < m; j++) { LL czy; czy = (LL)IsUsableCell(i, j); //cin >> czy; v[i + 1][j + 1] = czy; } } makepre(); LL res = 0; for(int i = 1; i <= n; i++) { for(int j = 1; j <= m; j++) { if(v[i][j] == 0) continue; LL x = solve(i, j); res += x; //cout << i << " " << j << " " << x << endl; } } printf("%lld", res); return 0; } |