Submission #1153373

#	Time	Username	Problem	Language	Result	Execution time	Memory
1153373		browntoad	Rectangles (IOI19_rect)	C++20	0 / 100	12 ms	23880 KiB

rect

#include <bits/stdc++.h>
#include "rect.h"
using namespace std;
#define ll long long
// #define int ll
#define FOR(i, a, b) for (int i = (a); i < (b); i++)
#define REP(i, n) FOR(i, 0, n)
#define RREP(i, n) for (int i = (n)-1; i >= 0; i--)
#define RREP1(i, n) for (int i = (n); i >= 1; i--)
#define REP1(i, n) FOR(i, 1, n+1)
#define pii pair<int, int>
#define ppi pair<pii, int>
#define pip pair<int, pii>
#define f first
#define s second
#define pb push_back
#define ALL(x) (x).begin(), (x).end()
#define SZ(x) (int)((x).size())
#define endl '\n'
#define IOS() ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)

const ll maxn = 705;
const ll inf = 1e9;
const ll mod = 998244353;

ll pw(ll x, ll p, ll m){
    ll ret = 1;
    while(p > 0){
        if (p & 1){
            ret *= x;
            ret %= m;
        }
        x *= x;
        x %= m;
        p >>= 1;
    }
    return ret;
}
ll inv(ll x, ll m){
    return pw(x, m-2, m);
}

int n, m;

vector<pii> rngr[maxn][maxn], rngd[maxn][maxn]; // given row, then column, the ok ranges
ll count_rectangles(vector<vector<int>> arr){
    n = SZ(arr);
    m = SZ(arr[0]);

    ll ret = 0;
    for (int i = 1; i <= n-2; i++){
        stack<pii> st;
        int lim = m+5;
        for (int j = m-1; j >= 0; j--){
            // subtask
            while(SZ(st)){
                if (st.top().s != j+1){
                    rngr[i][j+1].pb({j+1, st.top().s-1});

                    if (st.top().s-1 <= lim) ret++;
                }
                if (st.top().f > arr[i][j]) break;
                if (st.top().f == arr[i][j]) {
                    st.pop();
                    break;
                }
                st.pop();
            }

            st.push({arr[i][j], j});

            if (arr[i][j] >= arr[i+1][j] || arr[i][j] >= arr[i-1][j]) lim = j;
            //ret += SZ(rngr[i][j+1]);

        }
    }

    return ret;
}
/*
signed main(){
    cout<<count_rectangles({{4, 8, 7, 5, 6}, {7, 4, 10, 3, 5}, {9, 7, 20, 14, 2}})<<endl;
}
*/


/*
4 4
1 2
2 3
3 1
1 4
*/

• Subtask 1: 0 / 8
• Subtask 2: 0 / 7
• Subtask 3: 0 / 12
• Subtask 4: 0 / 22
• Subtask 5: 0 / 10
• Subtask 6: 0 / 13
• Subtask 7: 0 / 28