Submission #118261

#	Time	Username	Problem	Language	Result	Execution time	Memory
118261		Charis02	Bitaro the Brave (JOI19_ho_t1)	C++14	50 / 100	416 ms	274432 KiB

joi2019_ho_t1

#include<iostream>
#include<stdio.h>
#include<vector>
#include<cmath>
#include<queue>
#include<string.h>
#include<map>
#include<set>
#include<algorithm>
#define ll unsigned long long
#define pi pair < ll,ll >
#define mp(a,b) make_pair(a,b)
#define rep(i,a,b) for(int i = a;i < b;i++)
#define N 5004
#define INF 1e9+7

using namespace std;

ll n,m,ans;
ll orbs[N][N],jewels[N][N],ignots[N][N];
ll sumo[N][N],sumi[N][N];
char c;

ll get_orbs(ll si,ll sj,ll ei,ll ej)
{
    if(si == 0 && sj == 0)
        return sumo[ei][ej];
    else if(si == 0)
        return sumo[ei][ej] - sumo[ei][sj-1];
    else if(sj == 0)
        return sumo[ei][ej] - sumo[si-1][ej];

    return sumo[ei][ej] - sumo[si-1][ej] - sumo[ei][sj-1] + sumo[si-1][sj-1];
}

ll get_ignots(ll si,ll sj,ll ei,ll ej)
{
    if(si == 0 && sj == 0)
        return sumi[ei][ej];
    else if(si == 0)
        return sumi[ei][ej] - sumi[ei][sj-1];
    else if(sj == 0)
        return sumi[ei][ej] - sumi[si-1][ej];

    return sumi[ei][ej] - sumi[si-1][ej] - sumi[ei][sj-1] + sumi[si-1][sj-1];
}

int main()
{
    ios_base::sync_with_stdio(false);

    cin >> n >> m;

    rep(i,0,n)
    {
        rep(j,0,m)
        {
            cin >> c;

            if(c == 'J')
                jewels[i][j] = 1;
            else if(c == 'O')
                orbs[i][j] = 1;
            else
                ignots[i][j] = 1;
        }
    }

    rep(i,0,n)
    {
        rep(j,0,m)
        {
            if(i == 0 && j == 0)
            {
                sumi[i][j] = ignots[i][j];
                sumo[i][j] = orbs[i][j];
            }
            else if(i == 0)
            {
                sumi[i][j] = sumi[i][j-1]+ignots[i][j];
                sumo[i][j] = sumo[i][j-1]+orbs[i][j];
            }
            else if(j == 0)
            {
                sumi[i][j] = sumi[i-1][j]+ignots[i][j];
                sumo[i][j] = sumo[i-1][j]+orbs[i][j];
            }
            else
            {
                sumi[i][j] = sumi[i-1][j] - sumi[i-1][j-1] + sumi[i][j-1]+ ignots[i][j];
                sumo[i][j] = sumo[i-1][j]  - sumo[i-1][j-1] + sumo[i][j-1] + orbs[i][j];
            }
        }
    }

    rep(i,0,n)
    {
        rep(j,0,m)
        {
            if(i != n-1 && j != m-1 && jewels[i][j])
            {
                ans = (ans + get_ignots(i,j,n-1,j)*get_orbs(i,j,i,m-1));
            }
        }
    }

    cout << ans;

    return 0;
}

Compilation message (stderr)

joi2019_ho_t1.cpp: In function 'int main()':
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:54:9:
     rep(i,0,n)
         ~~~~~                       
joi2019_ho_t1.cpp:54:5: note: in expansion of macro 'rep'
     rep(i,0,n)
     ^~~
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:56:13:
         rep(j,0,m)
             ~~~~~                   
joi2019_ho_t1.cpp:56:9: note: in expansion of macro 'rep'
         rep(j,0,m)
         ^~~
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:69:9:
     rep(i,0,n)
         ~~~~~                       
joi2019_ho_t1.cpp:69:5: note: in expansion of macro 'rep'
     rep(i,0,n)
     ^~~
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:71:13:
         rep(j,0,m)
             ~~~~~                   
joi2019_ho_t1.cpp:71:9: note: in expansion of macro 'rep'
         rep(j,0,m)
         ^~~
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:96:9:
     rep(i,0,n)
         ~~~~~                       
joi2019_ho_t1.cpp:96:5: note: in expansion of macro 'rep'
     rep(i,0,n)
     ^~~
joi2019_ho_t1.cpp:13:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
joi2019_ho_t1.cpp:98:13:
         rep(j,0,m)
             ~~~~~                   
joi2019_ho_t1.cpp:98:9: note: in expansion of macro 'rep'
         rep(j,0,m)
         ^~~
joi2019_ho_t1.cpp:100:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             if(i != n-1 && j != m-1 && jewels[i][j])
                ~~^~~~~~
joi2019_ho_t1.cpp:100:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             if(i != n-1 && j != m-1 && jewels[i][j])
                            ~~^~~~~~

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