Submission #474252

#	Time	Username	Problem	Language	Result	Execution time	Memory
474252		Neos	Bob (COCI14_bob)	C++14	120 / 120	232 ms	17848 KiB

bob

#include <cstdio>
#include <stack>
#include <algorithm>
#include <cstring>

using namespace std;

const int MAXN = 1005;

int n, m, k;
int a[MAXN][MAXN];
int s[MAXN][MAXN];

typedef long long llint;

struct t{
  int cnt;
  int v, h;
  llint ms;
  t (int _cnt = 0, int _v = 0, int _h = 0, llint _ms = 0){
    cnt = _cnt;
    v = _v; h = _h;
    ms = _ms;
  }
};

llint ans;

int main (void){
  scanf("%d%d", &n, &m);
  for (int i = 0; i < n; ++i)
    for (int j = 0; j < m; ++j)
      scanf("%d", &a[i][j]);

  for (int i = 0; i < n; ++i){

    stack < t > S;
    S.push(t(0, -1, 0, 0));

    for (int j = 0; j < m; ++j){
      s[i][j] = 1;
      if (i > 0 && a[i-1][j] == a[i][j]) s[i][j] += s[i-1][j];

      int sum = 1;
      while (S.top().v == a[i][j] && S.top().h >= s[i][j]){
	sum += S.top().cnt;
	S.pop();
      }

      llint my_s = sum * (llint)s[i][j];
      if (S.top().v == a[i][j]) my_s += S.top().ms;
      ans += my_s;
      S.push(t(sum, a[i][j], s[i][j], my_s));
    }

  }

  printf("%lld\n", ans);


  return 0;
}

Compilation message (stderr)

bob.cpp: In function 'int main()':
bob.cpp:30:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |   scanf("%d%d", &n, &m);
      |   ~~~~~^~~~~~~~~~~~~~~~
bob.cpp:33:12: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |       scanf("%d", &a[i][j]);
      |       ~~~~~^~~~~~~~~~~~~~~~

• Subtask 1: 12 / 12
• Subtask 2: 12 / 12
• Subtask 3: 12 / 12
• Subtask 4: 12 / 12
• Subtask 5: 12 / 12
• Subtask 6: 12 / 12
• Subtask 7: 12 / 12
• Subtask 8: 12 / 12
• Subtask 9: 12 / 12
• Subtask 10: 12 / 12