Cod sursa(job #2792532)

Utilizator PatrickCplusplusPatrick Kristian Ondreovici PatrickCplusplus Data 1 noiembrie 2021 20:47:52
Problema Balans Scor 35
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.86 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("balans.in");
ofstream fout("balans.out");

const int nmax = 305;
int n, m, r, c, mat[nmax][nmax];
long long mat2[nmax][nmax], sum[nmax], sum2[nmax];

bool Check(int ans){
    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= m; ++j){
            mat2[i][j] = 1LL * mat[i][j] * 1000 - ans;
        }
    }
    for (int row = 1; row <= n; ++row){
        memset(sum, 0, sizeof sum);
        for (int row2 = row; row2 <= n; ++row2){
            for (int j = 1; j <= m; ++j){
                sum[j] += mat2[row2][j];
            }
            if (row2 - row + 1 >= r){
                for (int j = 1; j <= m; ++j){
                    sum2[j] = sum[j] + sum2[j - 1];
                }
                deque <int> dq;
                for (int j = 1; j <= m; ++j){
                    if (j >= c){
                        while (dq.size() > 0 && sum2[dq.back()] > sum2[j - c]) dq.pop_back();
                        dq.push_back(j - c);
                        if (sum2[j] - sum2[dq.front()] >= 0){
                            return true;
                        }
                    }
                }
            }
        }
    }
    return false;
}

int main(){
    fin >> n >> m >> r >> c;
    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= m; ++j){
            fin >> mat[i][j];
            int x = mat[i][j];
            mat[i + n][j] = x;
            mat[i][j + m] = x;
            mat[i + n][j + m] = x;
        }
    }
    n = n + n;
    m = m + m;
    int st = 1, dr = 100000000, ans;
    while (st <= dr){
        int mid = (st + dr) / 2;
        if (Check(mid)){
            ans = mid;
            st = mid + 1;
        }
        else{
            dr = mid - 1;
        }
    }
    fout << ans / 1000 << "." << ans % 1000;
    return 0;
}