snuke · July 29, 2018 14:24
diff --git a/hard.cpp b/hard.cpp
 modint calc(long long s) { // returns sum(S[0] ... S[pos(s)-1])
  modint res;
  while (n <= s) {
    res += sum(0, n);
    s -= n;
    s *= b;
  }
  res += sum(0, s);
  return res;
 }
 ans[0] = calc(s*b+1) - calc(s*b);
 ans[1] = calc(s*b) - calc(s*b-b);
	modint calc(long long s) { // returns sum(S[0] ... S[pos(s)-1])
	modint res;
	while (n <= s) {
	res += sum(0, n);
	s -= n;
	s *= b;
	}
	res += sum(0, s);
	return res;
	}
	ans[0] = calc(sb+1) - calc(sb);
	ans[1] = calc(sb) - calc(sb-b);