結果

問題 No.1201 お菓子配り-4
ユーザー SSRS
提出日時 2020-08-28 22:15:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,159 bytes
コンパイル時間 5,059 ms
コンパイル使用メモリ 194,524 KB
最終ジャッジ日時 2025-01-13 18:00:58
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 33 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#pragma GCC target("avx2")
#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1000000007;
//https://judge.yosupo.jp/submission/7028
long long gauss_sum(long long n, long long a, long long b, long long c) {
	if (n == 0) return 0;
	long long res = 0;
	{
		long long p = a / c;
		res += n * (n-1) / 2 * p;
		a %= c;
	}
	{
		long long p = b / c;
		res += n * p;
		b %= c;
	}
	if (a == 0) return res;
	long long top = (a * (n-1) + b) / c;
	res += top * n;
	long long h = 1;
	if (h <= top) {
		res -= gauss_sum(top - h + 1, c, c * h - (b + 1), a) + top - h + 1;
	}
	return res;
}
int main(){
  int N;
  cin >> N;
  int M;
  cin >> M;
  vector<long long> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  vector<long long> B(M);
  for (int i = 0; i < M; i++){
    cin >> B[i];
  }
  long long ans = 0;
  for (int i = 0; i < N; i++){
    for (int j = 0; j < M; j++){
      if (A[i] >= B[j]){
        long long tmp = B[j] * (B[j] + 1) % MOD * (A[i] / B[j]) % MOD;
        ans += tmp;
      }
      long long tmp = gauss_sum(B[j], A[i] % B[j], A[i] % B[j], B[j]);
      ans += tmp * 2 % MOD;
    }
  }
  ans %= MOD;
  cout << ans << endl;
}
0