Submission #1037532

#	Time	Username	Problem	Language	Result	Execution time	Memory
1037532		vannnnnnnn	Collecting Mushrooms (NOI18_collectmushrooms)	C++14	100 / 100	43 ms	37056 KiB

mushrooms

#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
using namespace std;
signed main()
{
  int r, c, d, p, ans=0;
  scanf("%lld %lld %lld %lld", &r, &c, &d, &p);
  char s[r+5][c+5];
  int pref[r+5][c+5];
  vector<pair<int, int> >v;
  memset(pref, 0, sizeof(pref));
  for(int i=1; i<=r; i++)
  {
    for(int j=1; j<=c; j++)
    {
      scanf(" %c", &s[i][j]);
//      printf("%c\n", s[i][j]);
      if(s[i][j]=='M')
      {
        v.push_back({i, j});
      }
      if(s[i][j]=='S')pref[i][j]++;
      pref[i][j]+=(pref[i-1][j]+pref[i][j-1]);
      pref[i][j]-=pref[i-1][j-1];
    }
  }
//  for(int i=1; i<=r; i++)
//  {
//    for(int j=1; j<=c; j++)
//    {
//      printf("%d ", pref[i][j]);
//    }
//    printf("\n");
//  }
  for(int i=0; i<v.size(); i++)
  {
//    printf("%lld\n", pref[min(r, v[i].fi+d)][min(c, v[i].se+d)]-pref[max(0ll, v[i].fi-d-1)][min(c, v[i].se+d)]-pref[min(r, v[i].fi+d)][max(0ll, v[i].se-d-1)]+pref[max(0ll, v[i].fi-d-1)][max(0ll, v[i].se-d-1)]);
    if(pref[min(r, v[i].fi+d)][min(c, v[i].se+d)]-pref[max(0ll, v[i].fi-d-1)][min(c, v[i].se+d)]-pref[min(r, v[i].fi+d)][max(0ll, v[i].se-d-1)]+pref[max(0ll, v[i].fi-d-1)][max(0ll, v[i].se-d-1)]>=p)
    {
      ans++;
    }
  }
  printf("%lld\n", ans);
}

Compilation message (stderr)

mushrooms.cpp: In function 'int main()':
mushrooms.cpp:37:17: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |   for(int i=0; i<v.size(); i++)
      |                ~^~~~~~~~~
mushrooms.cpp:9:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    9 |   scanf("%lld %lld %lld %lld", &r, &c, &d, &p);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
mushrooms.cpp:18:12: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   18 |       scanf(" %c", &s[i][j]);
      |       ~~~~~^~~~~~~~~~~~~~~~~

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