Submission #1096499

#	Time	Username	Problem	Language	Result	Execution time	Memory
1096499		lmToT27	Bitaro the Brave (JOI19_ho_t1)	C++14	100 / 100	60 ms	25008 KiB

joi2019_ho_t1

#include <bits/stdc++.h>
#define all(dataStructure) dataStructure.begin(),dataStructure.end()
#define ll long long

using namespace std;
namespace std {
        template <typename T, int D>
        struct _vector : public vector <_vector <T, D - 1>> {
                static_assert(D >= 1, "Dimension must be positive!");
                template <typename... Args>
                _vector(int n = 0, Args... args) : vector <_vector <T, D - 1>> (n, _vector <T, D - 1> (args...)) {}
        };
        // _vector <int, 3> a(n, m, k);: int a[n][m][k].
        // _vector <int, 3> a(n, m, k, x);: int a[n][m][k] initialized with x.
        template <typename T>
        struct _vector <T, 1> : public vector <T> {
                _vector(int n = 0, const T& val = T()) : vector <T> (n, val) {}
        };
}

const int MAX = 3e3 + 3;
const ll MOD[] = {1000000007, 998244353};

int n, m;
string st[MAX];
int r[MAX][256], c[MAX][256];


void Solve() {
        cin >> n >> m;
        for (int i = 0; i < n; i++) {
                cin >> st[i];
        }
        ll res = 0;
        for (int i = n - 1; i >= 0; i--) {
                for (int j = m - 1; j >= 0; j--) {
                        r[i][st[i][j]]++;
                        c[j][st[i][j]]++;
                        if (st[i][j] == 'J') res += r[i]['O'] * c[j]['I'];
                }
        }
        cout << res;
}

int32_t main() {
        ios_base::sync_with_stdio(0);
        cin.tie(0);

        #define TASK "TASK"
        if (fopen(TASK".INP", "r")) {
                freopen(TASK".INP", "r", stdin);
                freopen(TASK".OUT", "w", stdout);
        }
        
        /* int TEST = 1; cin >> TEST; while (TEST--) */ Solve();

        cerr << "\nTime elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << "s\n";
        return 0;
}

Compilation message (stderr)

joi2019_ho_t1.cpp: In function 'void Solve()':
joi2019_ho_t1.cpp:37:38: warning: array subscript has type 'char' [-Wchar-subscripts]
   37 |                         r[i][st[i][j]]++;
      |                                      ^
joi2019_ho_t1.cpp:38:38: warning: array subscript has type 'char' [-Wchar-subscripts]
   38 |                         c[j][st[i][j]]++;
      |                                      ^
joi2019_ho_t1.cpp: In function 'int32_t main()':
joi2019_ho_t1.cpp:51:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   51 |                 freopen(TASK".INP", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
joi2019_ho_t1.cpp:52:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   52 |                 freopen(TASK".OUT", "w", stdout);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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