結果

問題 No.886 Direct
ユーザー QCFiumQCFium
提出日時 2019-08-20 11:14:25
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,692 ms / 4,000 ms
コード長 3,231 bytes
コンパイル時間 1,987 ms
コンパイル使用メモリ 172,808 KB
実行使用メモリ 15,104 KB
最終ジャッジ日時 2024-04-16 02:01:40
合計ジャッジ時間 14,925 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 9 ms
5,376 KB
testcase_19 AC 7 ms
5,376 KB
testcase_20 AC 4 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 9 ms
5,376 KB
testcase_23 AC 218 ms
8,832 KB
testcase_24 AC 515 ms
9,344 KB
testcase_25 AC 192 ms
6,912 KB
testcase_26 AC 158 ms
8,064 KB
testcase_27 AC 924 ms
14,208 KB
testcase_28 AC 940 ms
13,824 KB
testcase_29 AC 37 ms
15,104 KB
testcase_30 AC 36 ms
15,104 KB
testcase_31 AC 1,692 ms
15,104 KB
testcase_32 AC 1,663 ms
14,976 KB
testcase_33 AC 1,658 ms
14,976 KB
testcase_34 AC 1,656 ms
15,104 KB
testcase_35 AC 1,660 ms
15,104 KB
権限があれば一括ダウンロードができます

ソースコード

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();
		int prod[1 << n_fact];
		for (int j = 0; j < n_fact; j++) prod[1 << j] = facts[j];
		prod[0] = 1;
		for (int j = 1; j < 1 << n_fact; j++) prod[j] = prod[j & j - 1] * prod[j & ~(j - 1)];
		mint so_num = 0;
		mint so_sum = 0;
		for (int i = 0; i < 1 << n_fact; i++) {
			int num = (h - 1) / prod[i];
			mint sum = mint(num) * (num + 1) / 2 * prod[i];
			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