結果

問題 No.1680 Sum and Difference
コンテスト
ユーザー ok
提出日時 2021-09-17 22:25:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,141 bytes
コンパイル時間 663 ms
コンパイル使用メモリ 80,128 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-29 21:04:20
合計ジャッジ時間 1,447 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<iomanip>
#include<cmath>
#include<vector>
#include<algorithm>
#include<utility>

using namespace std;

#define int long long
#define endl "\n"

constexpr long long INF = (long long)1e18;
constexpr long long MOD = 1000000007; 

struct fast_io {
	fast_io(){
		std::cin.tie(nullptr);
		std::ios::sync_with_stdio(false);
	};
} fio;

long long power(long long x, long long n){
	long long ans = 1;
	for(;n;n >>= 1, x *= x, ans %= MOD, x %= MOD) if(n&1)ans*=x;
	return ans % MOD;
}

int solve(int A, int B){
	int res = 0;
	int i1 = (A - B) / 2, i2 = (A + B) / 2;
	
	i1 %= MOD;
	i2 %= MOD;
	A %= MOD;
	B %= MOD;
	
	res += (B + 1 + B + i1)%MOD *(i1)%MOD*power(2, MOD-2)%MOD;
	res %= MOD;
	res += (A - (i1 + 1) + A - i2)%MOD * (i2 - i1)%MOD*power(2, MOD-2)%MOD;
	res %= MOD;
	res -= (MOD-B + 1 +MOD- B + i2)%MOD * (i2 - 0)%MOD*power(2, MOD-2)%MOD;
	res += i2;
	res = (res%MOD + MOD) % MOD;
	res *= 2;
	res += B * 2 + 1;
	res %= MOD;
	
	return res;
}

signed main(){
	cout<<fixed<<setprecision(10);
	
	int A, B;
	int ans = 0;
	
	cin>>A>>B;
	
	if(B > A) swap(A, B);
	
	cout<<solve(A, B)<<endl;
	
	return 0;
}
0