Submission #1151661

#	Time	Username	Problem	Language	Result	Execution time	Memory
1151661		dobri_oke	Collecting Mushrooms (NOI18_collectmushrooms)	C++20	100 / 100	15 ms	13128 KiB

mushrooms

//#pragma GCC target ("avx2")   
//#pragma GCC optimize ("Ofast")   
#include <bits/stdc++.h>   
using namespace std;  
#define int long long  
#define F first 
#define S second 
#define pb push_back 
const int N = 1e3+100, NN=26;
const int mod=1e9+7;
//int gcd(int a, int b) { if (b == 0) return a; else return gcd(b, a % b); }
//int lcm(int a, int b) { return a / gcd(a, b) * b; }
//int binpow(int a,int b){if(!b)return 1; if(b&1)return a*binpow(a,b-1)%mod; int x=binpow(a,b/2); return x*x%mod;}
signed main(){   
    ios_base::sync_with_stdio(0);   
    cin.tie(0);   
    cout.tie(0);
    int r, c, d, k;
    cin >> r >> c >> d >> k;
    char a[r+1][c+1];
    int p[r+2][c+2]{};
    for(int i=1;i<=r;i++){
        for(int j=1;j<=c;j++) cin >> a[i][j];
    }
    for(int i=1;i<=r;i++){
        for(int j=1;j<=c;j++){
            if(a[i][j]=='S'){
                p[max(i-d, 1ll)][max(j-d, 1ll)]++;
                p[min(i+d, r)+1][max(j-d, 1ll)]--;
                p[max(i-d, 1ll)][min(j+d, c)+1]--;
                p[min(i+d, r)+1][min(j+d, c)+1]++;
            }
        }
    }
    for(int i=1;i<=r;i++){
        for(int j=1;j<=c;j++){
            p[i][j]+=p[i][j-1];
        }
    }
    for(int i=1;i<=c;i++){
        for(int j=1;j<=r;j++){
            p[j][i]+=p[j-1][i];
        }
    }
    int ans=0;
    for(int i=1;i<=r;i++){
        for(int j=1;j<=c;j++){
            if(a[i][j]=='M' && p[i][j]>=k) ans++;
        }
    }
    cout << ans;
}

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