Submission #713743

#	Time	Username	Problem	Language	Result	Execution time	Memory
713743		ALTAKEXE	Bob (COCI14_bob)	C++14	120 / 120	126 ms	17888 KiB

bob

#include <cstdio>
#include <stack>
#include <algorithm>
#include <cstring>
using namespace std;
int n, m, k;
int a[1005][1005];
int s[1005][1005];
struct t{
  int cnt;
  int v, h;
  long long  ms;
  t (int _cnt = 0, int _v = 0, int _h = 0, long long  _ms = 0){
    cnt = _cnt;
    v = _v; h = _h;
    ms = _ms;
  }
};
int main (){
  long long ans=0;
  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();
      }
      long long  my_s = sum * (long long )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", ans);
  return 0;
}

Compilation message (stderr)

bob.cpp: In function 'int main()':
bob.cpp:21:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   21 |   scanf("%d %d", &n, &m);
      |   ~~~~~^~~~~~~~~~~~~~~~~
bob.cpp:24:12: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   24 |       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