Comet OJ - Contest #4--前缀和

 

原题：Comet OJ - Contest #4-B

https://www.cometoj.com/contest/39/problem/B?problem_id=1577传送门

 

一开始就想着暴力打表，结果。。

前缀和是个很好的工具，本题可以用相邻前缀和之差得到结果。

例如：K=4：

1	1	1	1
0	1	1	1
1	0	1	1
1	1	0	1

 

#include <cstdio>

using int64 = long long;

int main() {
  int T;
  scanf("%d", &T);
  for (int cas = 1; cas <= T; ++cas) {
    int64 l, r, k;
    scanf("%lld%lld%lld", &l, &r, &k);
    if (k & 1) printf("%lld\n", r - l + 1);
    else {
      auto sum = [] (int64 n, int64 k) {
        if (n < k) return n + 1;
        int64 l = n / k;
        int64 zeros = l / (k + 1) * k;
        if (l % (k + 1)) {
          zeros += l % (k + 1) - 1;
          zeros += n % k >= l % (k + 1) - 1;
        }
        return n + 1 - zeros;
      };
      printf("%lld\n", sum(r, k) - sum(l - 1, k));
    }
  }
  return 0;
}