Submission #689297

#	Time	Username	Problem	Language	Result	Execution time	Memory
689297		zeroesandones	Bitaro the Brave (JOI19_ho_t1)	C++17	100 / 100	280 ms	159816 KiB

joi2019_ho_t1

#include "bits/stdc++.h"
using namespace std;

typedef long long ll;
typedef long double ld;
typedef vector<ll> vi;
typedef pair<ll, ll> pi;

#define FOR(i, j, k) for (ll i = j; i < (ll) k; ++i)
#define FORD(i, j, k) for (ll i = j; i >= (ll) k; --i)
#define nl "\n"
#define sp " "

#define all(x) (x).begin(), (x).end()
#define sc second
#define fr first
#define pb emplace_back

void solve()
{
    ll n, m;
    cin >> n >> m;

    string a[n];
    FOR(i, 0, n) {
        cin >> a[i];
    }

    ll ans = 0;

    ll row[n][m] = {};
    ll col[n][m] = {};
    FOR(i, 0, n) {
        FOR(j, 0, m) {
            row[i][j] = (a[i][j] == 'O');
            if(j) row[i][j] += row[i][j - 1];
        }
    }

    FOR(j, 0, m) {
        FOR(i, 0, n) {
            col[i][j] = (a[i][j] == 'I');
            if(i) col[i][j] += col[i - 1][j];
        }
    }

    FOR(i, 0, n) {
        FOR(j, 0, m) {
            if(a[i][j] != 'J')
                continue;
            ll O = row[i][m - 1] - row[i][j];
            ll I = col[n - 1][j] - col[i][j];
            ans += (O * I);
        }
    }

    cout << ans << nl;
}

signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    ll t = 1;
    // cin >> t;
    while (t--)
    {
        solve();
    }
}

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50