結果

問題 No.886 Direct
コンテスト
ユーザー niuez
提出日時 2019-09-14 09:05:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 297 ms / 4,000 ms
コード長 2,371 bytes
コンパイル時間 1,582 ms
コンパイル使用メモリ 173,092 KB
実行使用メモリ 75,776 KB
最終ジャッジ日時 2024-07-05 08:38:54
合計ジャッジ時間 5,488 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
#define rep(i,s,e) for(i64 (i) = (s);(i) < (e);(i)++)
#define all(x) x.begin(),x.end()
#define let auto const

template<typename... Types>
struct dynarr: std::vector<Types...> {
  using std::vector<Types...>::vector;
  using size_type = typename std::vector<Types...>::size_type;
  auto&& operator[](size_type i) { return this->at(i); }
  auto&& operator[](size_type i) const { return this->at(i); }
};

template<i64 M>
struct modint {
  i64 a;
  constexpr modint(const i64 x = 0) noexcept: a((x % M + M) % M) {}
  constexpr i64 value() const noexcept { return a; }
  constexpr modint& operator+=(const modint r) noexcept {
    a += r.a;
    if(a >= M) a -= M;
    return *this;
  }
  constexpr modint& operator=(const i64 r) {
    a = (r % M + M) % M;
    return *this;
  }
  constexpr modint& operator-=(const modint r) noexcept {
    a -= r.a;
    if(a < 0) a += M;
    return *this;
  }
  constexpr modint& operator*=(const modint r) noexcept {
    a = a * r.a % M;
    return *this;
  }
  constexpr modint& operator/=(modint r) noexcept {
    i64 ex = M - 2;
    while(ex) {
      if(ex & 1) {
        *this *= r;
      }
      r *= r;
      ex >>= 1;
    }
    return *this;
  }

  constexpr modint operator+(const modint r) const {
    return modint(*this) += r;
  }
  constexpr modint operator-(const modint r) const {
    return modint(*this) -= r;
  }
  constexpr modint operator*(const modint r) const {
    return modint(*this) *= r;
  }
  constexpr modint operator/(const modint r) const {
    return modint(*this) /= r;
  }
};

const i64 mod = 1e9 + 7;

using fp = modint<mod>;


int main() {
  i64 H, W;
  cin >> H >> W;
  i64 N = max(H, W) + 1;
  vector<fp> a(N), b(N);
  rep(i,1,H) a[i] = fp(H - i);
  rep(i,1,W) b[i] = fp(W - i);

  vector<i64> sieve(N, 1);
  vector<i64> prime;
  rep(p,2,N) {
    if(sieve[p]) {
      prime.push_back(p);
      for(i64 k = 1;k * p < N; ++k) {
        sieve[k * p] = false;
      }
    }
  }

  for(auto p: prime) {
    for(i64 i = (N - 1) / p; i > 0; i--) {
      a[i] += a[i * p];
      b[i] += b[i * p];
    }
  }
  for(i64 i = 1;i < N;i++) a[i] *= b[i];
  for(auto p: prime) {
    i64 m = (N - 1) / p;
    for(i64 i = 1; i <= m; i++) {
      a[i] -= a[i * p];
    }
  }

  cout << (a[1] * fp(2) + fp(H * (W - 1) + W * (H - 1))).value() << endl;
}
0