Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include<bits/stdc++.h>
- using namespace std;
- int n, m, k;
- int a[5010][5010];
- int b[5010][5010];
- int main(){
- ios_base::sync_with_stdio(false);
- cin.tie(0);cout.tie(0);
- cin >> n >> m >> k;
- for(int i=1; i<=n; i++){
- for(int j=1; j<=m; j++){
- cin >> a[i][j];
- }
- }
- for(int i=1; i<=n; i++){
- deque<int> dq;
- for(int j=1; j<=m; j++) {
- while (dq.size() && a[i][dq.back()]<=a[i][j]) dq.pop_back();
- dq.push_back(j);
- while (dq.size() && j-dq.front()+1>k) dq.pop_front();
- if (j>=k) b[i][j]=a[i][dq.front()];
- }
- }
- memset(a,0,sizeof a);
- for(int i=k; i<=m; i++){
- deque<int> dq;
- for(int j=1; j<=n; j++){
- while (dq.size() && b[dq.back()][i]<=b[j][i]) dq.pop_back();
- dq.push_back(j);
- while (dq.size() && j-dq.front()+1>k) dq.pop_front();
- if (j>=k) a[j][i]=b[dq.front()][i];
- }
- }
- long long ans=0;
- for(int i=k; i<=n; i++){
- for(int j=k; j<=m; j++){
- ans += a[i][j];
- }
- }
- cout << ans;
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement