Kangaroo Keeping

#include <bits/stdc++.h>
using namespace std;
const int SN = 317;
long cs[SN];
long sm[SN*SN], smb[SN];
int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  int n, m, q;
  cin >> n >> m >> q;
  for (int qi = 0; qi < q; ++qi) {
    int t;
    cin >> t;
    if (t == 1) {
      int s;
      cin >> s;
      if (s < SN) cs[s]++;
      else {
        for (int i = s; i <= n; i += s) {
          sm[i]++;
          smb[i/SN]++;
        }
      }
    } else {
      int a, b;
      cin >> a >> b;
      long sum = 0;
      for (int i = 1; i < SN; ++i) {
        int xa = a/i+!!(a%i);
        int xb = b/i;
        if (xa > xb) continue;
        sum += (xb-xa+1)*cs[i];
      }
      for (int i = a; i <= b; ++i) {
        if (i%SN || i+SN-1 > b) {
          sum += sm[i];
        } else {
          sum += smb[i/SN];
          i += SN-1;
        }
      }
      cout << sum << "\n";
    }
  }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED