Submission #621195

#	Time	Username	Problem	Language	Result	Execution time	Memory
621195		srivatsav_kannan	Collecting Mushrooms (NOI18_collectmushrooms)	C++14	38 / 100	41 ms	9572 KiB

mushrooms

#include <iostream>
#include <iomanip>
#include <array>
#include <fstream>
#include <vector>
#include <set>
#include <queue>
#include <cmath>
#include <map>
#include <algorithm>
#include <numeric>
#include <stack>
#include <cstring>
#include <bitset>
#include <climits>
#include <valarray>
#include <list>
#include <functional>
#include <unordered_map>
#define int long long int
#define inf 100000000000
#define endl '\n'
#define mod 998244353
using namespace std;
signed main(){
    int r,c,d,k; cin >> r >> c >> d >> k;
    char ar[r+1][c+1];
    int val[r+1][c+1];
    for (int i = 1; i <= r; i++){
        for (int j = 1; j <= c; j++){
            cin >> ar[i][j];
            val[i][j] = 0;
        }
    }
    for (int i = 0; i <= r; i++) val[i][0] = 0;
    for (int i = 0; i <= c; i++) val[0][i] = 0;
    
    for (int i = 1; i <= r; i++){
        for (int j = 1; j <= c; j++){
            if (ar[i][j] != 'S') continue;
            int ci = max(i-d,1LL);
            int cj = max(j-d,1LL);
            if (ci >= 1 && cj >= 1) val[ci][cj]++;
            ci = max(i-d, 1LL);
            cj = j+d+1;
            if (ci >= 1 && cj <= c) val[ci][cj]--;
            ci = i+d+1;
            cj = j+d;
            if (ci <= r && cj >= 1) val[ci][cj]--;
//            ci = min(i+d+1, r);
//            cj = min(j+d+1, c);
//            if (ci <= r && cj <= r) val[ci][cj]++;
        }
    }
    for (int i = 1; i <= r; i++){
        for (int j = 1; j <= c; j++){
            val[i][j] += val[i-1][j];
            val[i][j] += val[i][j-1];
            val[i][j] -= val[i-1][j-1];
            //cout << val[i][j] << " ";
        }
        //cout << endl;
    }
    
    int cnt = 0;
    for (int i = 1; i <= r; i++){
        for (int j = 1; j <= c; j++){
            //if (i == 3 && j == 4) cout << ar[i][j] << " " << val[i][j] << endl;
            if (ar[i][j] == 'M' && val[i][j] >= k) cnt++;
        }
    }
    cout << cnt << endl;
}

• Subtask 1: 9 / 9
• Subtask 2: 10 / 10
• Subtask 3: 0 / 18
• Subtask 4: 0 / 23
• Subtask 5: 19 / 19
• Subtask 6: 0 / 21