Submission #1231212

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

#define REP(i,n) for(int i=0,_n=(int)(n); i < _n; i++)
template<class T> bool chmax(T &a, T b) { return a < b ? (a = b, true) : false; }
template<class T> bool chmin(T &a, T b) { return a > b ? (a = b, true) : false; }

typedef long long ll;

const int MAX_N = 100000 + 10;
int R, C, K, N;
pair<int,int> rc[MAX_N];
int col[100000 + 10];
int rcol[100000 + 10];

int main2() {
  memset(col, 0, sizeof(col));
  memset(rcol, 0, sizeof(rcol));

  scanf("%d%d%d", &R, &C, &K);
  scanf("%d", &N);
  REP(i, N) {
    int r, c;
    scanf("%d%d", &r, &c);
    rc[i] = {r, c};
  }
  sort(rc, rc + N);

  // 以下のようにして、各行に対して O(1) 程度で数える。
  // col[c] により列 c にいくつの点があるかを保持する。
  // rcol[t] により 点をt個持つ列の個数を保持する。
  // 今見ている行にある点を除く（ col[c], rcol[t] を操作する )
  // t = K - [今見ている行にある点の個数] として rcol[t] を答えに足す
  // 今見ていた行にある点を戻す （ col[c], rcol[t] を操作する )

  rcol[0] = C;
  REP(i, N) {
    int c = rc[i].second;
    --rcol[ col[c] ];
    ++col[c];
    ++rcol[ col[c] ];
  }

  ll ans = 0;

  int sp = 0;
  for (int r = 1; r <= R; r++) {
    int len = 0;
    while (sp + len < N && rc[sp + len].first == r) len++;
    
    for (int i = sp; i < sp + len; i++) {
      int c = rc[i].second;
      --rcol[ col[c] ];
      --col[c];
      ++rcol[ col[c] ];
    }

    int t = K - len; 
    if (t >= 0)
      ans += rcol[t];

    for (int i = sp; i < sp + len; i++) {
      int c = rc[i].second;
      --rcol[ col[c] ];
      ++col[c];
      ++rcol[ col[c] ];
    }

    sp = sp + len;
  }

  cout << ans << endl;

  return 0;
}

int main() {
  for (;!cin.eof();cin>>ws) main2();
  return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main2()’:
./Main.cpp:20:30: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d%d", &R, &C, &K);
                              ^
./Main.cpp:21:18: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d", &N);
                  ^
./Main.cpp:24:26: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &r, &c);
                          ^

Judge Result