結果

問題 No.886 Direct
コンテスト
ユーザー QCFium
提出日時 2019-08-19 22:50:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,510 ms / 4,000 ms
コード長 3,120 bytes
コンパイル時間 1,946 ms
コンパイル使用メモリ 177,220 KB
実行使用メモリ 15,104 KB
最終ジャッジ日時 2024-12-15 03:52:04
合計ジャッジ時間 13,130 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define MOD 1000000007

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

template<int mod>
struct ModInt{
    int x;
    ModInt():x(0){}
    ModInt(long long y):x(y>=0?y%mod:(mod-(-y)%mod)%mod){}
    ModInt &operator+=(const ModInt &p){
        if((x+=p.x)>=mod)x-=mod;
        return *this;
    }
    ModInt &operator-=(const ModInt &p){
        if((x+=mod-p.x)>=mod)x-=mod;
        return *this;
    }
    ModInt &operator*=(const ModInt &p){
        x=(int)(1LL*x*p.x%mod);
        return *this;
    }
    ModInt &operator/=(const ModInt &p){
        *this*=p.inverse();
        return *this;
    }
	ModInt &operator^=(long long p){
		ModInt res = 1;
		for (; p; p >>= 1) {
			if (p & 1) res *= *this;
			*this *= *this;
		}
		return *this = res;
	}
    ModInt operator-()const{return ModInt(-x);}
    ModInt operator+(const ModInt &p)const{return ModInt(*this)+=p;}
    ModInt operator-(const ModInt &p)const{return ModInt(*this)-=p;}
    ModInt operator*(const ModInt &p)const{return ModInt(*this)*=p;}
    ModInt operator/(const ModInt &p)const{return ModInt(*this)/=p;}
	ModInt operator^(long long p)const{return ModInt(*this)^=p;}
    bool operator==(const ModInt &p)const{return x==p.x;}
    bool operator!=(const ModInt &p)const{return x!=p.x;}
    explicit operator int() const { return x; }                           // added by QCFium
    ModInt operator=(const int p) {x = p; return ModInt(*this);} // added by QCFium
    ModInt inverse()const{
        int a=x,b=mod,u=1,v=0,t;
        while(b>0){
            t=a/b;
            a-=t*b;
            std::swap(a,b);
            u-=t*v;
            std::swap(u,v);
        }
        return ModInt(u);
    }
    friend std::ostream &operator<<(std::ostream &os,const ModInt<mod> &p){
        return os<<p.x;
    }
    friend std::istream &operator>>(std::istream &is,ModInt<mod> &a){
        long long x;
        is>>x;
        a=ModInt<mod>(x);
        return (is);
    }
};
typedef ModInt<MOD> mint;

mint solve(int h, int w) {
	if (h < w) std::swap(h, w);
	
	std::vector<int> pf_table(h + 1);
	for (int i = 2; i <= h; i++)
		if (!pf_table[i]) for (int j = i; j <= h / i; j++) pf_table[j * i] = i;
	
	mint res = mint(h) * (w - 1) + mint(w) * (h - 1);
	for (int i = 1; i < w; i++) {
		std::vector<int> facts;
		for (int cur = i; ; ) {
			if (!pf_table[cur]) {
				if (cur > 1) facts.push_back(cur);
				break;
			}
			facts.push_back(pf_table[cur]);
			cur /= pf_table[cur];
		}
		std::sort(facts.begin(), facts.end());
		facts.erase(std::unique(facts.begin(), facts.end()), facts.end());
		
		int n_fact = facts.size();
		mint so_num = 0;
		mint so_sum = 0;
		for (int i = 0; i < 1 << n_fact; i++) {
			int cur = 1;
			for (int j = 0; j < n_fact; j++) if (i >> j & 1) cur *= facts[j];
			int num = (h - 1) / cur;
			mint sum = mint(num) * (num + 1) / 2 * cur;
			if (__builtin_popcount(i) & 1) so_num -= num, so_sum -= sum;
			else so_num += num, so_sum += sum;
		}
		res += mint(2) * (w - i) * (so_num * h - so_sum);
	}
	
	return res;
}

int main() {
	int h = ri(), w = ri();
	std::cout << solve(h, w) << std::endl;
	return 0;
}
0