Submission #624819

#TimeUsernameProblemLanguageResultExecution timeMemory
624819QwertyPiSandcastle 2 (JOI22_ho_t5)C++14
15 / 100
5028 ms50340 KiB
#include <bits/stdc++.h> #define int long long using namespace std; const int SQ = 240; int a[200013]; int b[200013]; bool f[200013]; int s[200013]; int h, w; int id(int i, int j){ return i * (w + 1) + j; } int id2(int i, int j){ return j * (h + 1) + i; } struct Range{ int x1, x2, y1, y2; // 1D; [x1, x2] x [y1, y2] }; vector<Range> vrs[SQ][SQ]; int qry(int x1, int x2, int y1, int y2){ return s[id(x2 + 1, y2 + 1)] + s[id(x1, y1)] - s[id(x2 + 1, y1)] - s[id(x1, y2 + 1)]; } int k[1501][1501]; int g(vector<Range> vr, int h){ for(int i = 0; i < w; i++){ for(int j = 0; j < w; j++){ k[i][j] = 0; } } for(auto r : vr){ for(int a = r.x1; a <= r.x2; a++){ for(int b = r.y1; b <= r.y2; b++){ k[a][b]++; } } } int ans = 0; for(int i = 0; i < w; i++){ for(int j = i; j < w; j++){ ans += k[i][j] == h * (j - i + 1) - 1; } } return ans; } int32_t main(){ cin >> h >> w; for(int i = 0; i < h; i++){ for(int j = 0; j < w; j++){ cin >> a[id(i, j)]; } } if(h > w){ for(int i = 0; i < h; i++){ for(int j = 0; j < w; j++){ b[id2(i, j)] = a[id(i, j)]; } } for(int i = 0; i < h; i++){ for(int j = 0; j < w; j++){ a[id2(i, j)] = b[id2(i, j)]; } } swap(w, h); } { set<int> S; map<int, int> M; for(int i = 0; i < h; i++){ for(int j = 0; j < w; j++){ S.insert(a[id(i, j)]); } } int idx = 0; for(auto j : S){ M[j] = idx++; } for(int i = 0; i < h; i++){ for(int j = 0; j < w; j++){ a[id(i, j)] = M[a[id(i, j)]]; } } } int ans = 0; for(int i = 0; i < h; i++){ for(int j = 0; j < w - 1; j++){ int v1 = a[id(i, j)], v2 = a[id(i, j + 1)]; if(v1 > v2) swap(v1, v2); for(int x = 0; x < h; x++){ for(int y = 0; y < w; y++){ f[id(x, y)] = a[id(x, y)] > v1 && a[id(x, y)] < v2; } } for(int x = 0; x < h; x++){ for(int y = 0; y < w; y++){ s[id(x + 1, y + 1)] = s[id(x + 1, y)] + s[id(x, y + 1)] - s[id(x, y)] + f[id(x, y)]; } } int x1 = i, x2 = i, y1 = j, y2 = j + 1; for(int x3 = 0; x3 <= x1; x3++){ for(int x4 = x2; x4 < h; x4++){ if(qry(x3, x4, y1, y2)) continue; int y3 = INT32_MAX, y4 = INT32_MIN; { int l = 0, r = y1; while(l != r){ int mid = (l + r) / 2; if(qry(x3, x4, mid, y1)){ l = mid + 1; }else{ r = mid; } } y3 = l; } { int l = y2, r = w - 1; while(l != r){ int mid = (l + r + 1) / 2; if(qry(x3, x4, y2, mid)){ r = mid - 1; }else{ l = mid; } } y4 = l; } vrs[x3][x4].push_back({y3, y1, y2, y4}); } } } } for(int i = 0; i < h - 1; i++){ for(int j = 0; j < w; j++){ int v1 = a[id(i, j)], v2 = a[id(i + 1, j)]; if(v1 > v2) swap(v1, v2); for(int x = 0; x < h; x++){ for(int y = 0; y < w; y++){ f[id(x, y)] = a[id(x, y)] > v1 && a[id(x, y)] < v2; } } for(int x = 0; x < h; x++){ for(int y = 0; y < w; y++){ s[id(x + 1, y + 1)] = s[id(x + 1, y)] + s[id(x, y + 1)] - s[id(x, y)] + f[id(x, y)]; } } int x1 = i, x2 = i + 1, y1 = j, y2 = j; for(int x3 = 0; x3 <= x1; x3++){ for(int x4 = x2; x4 < h; x4++){ if(qry(x3, x4, y1, y2)) continue; int y3 = INT32_MAX, y4 = INT32_MIN; { int l = 0, r = y1; while(l != r){ int mid = (l + r) / 2; if(qry(x3, x4, mid, y1)){ l = mid + 1; }else{ r = mid; } } y3 = l; } { int l = y2, r = w - 1; while(l != r){ int mid = (l + r + 1) / 2; if(qry(x3, x4, y2, mid)){ r = mid - 1; }else{ l = mid; } } y4 = l; } vrs[x3][x4].push_back({y3, y1, y2, y4}); } } } } for(int x1 = 0; x1 < h; x1++){ for(int x2 = x1; x2 < h; x2++){ ans += g(vrs[x1][x2], x2 - x1 + 1); } } cout << ans << endl; }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...