結果

問題 No.1201 お菓子配り-4
ユーザー SSRS
提出日時 2020-08-28 21:58:45
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 966 bytes
コンパイル時間 3,787 ms
コンパイル使用メモリ 214,844 KB
最終ジャッジ日時 2025-01-13 17:27:34
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35 TLE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:1:28: warning: bad option ‘-favx2’ to pragma ‘optimize’ [-Wpragmas]
    1 | #pragma GCC optimize("avx2")
      |                            ^
main.cpp:6:81: warning: bad option ‘-favx2’ to attribute ‘optimize’ [-Wattributes]
    6 | long long sum_of_floor_linear(long long a, long long b, long long c, long long n){
      |                                                                                 ^
main.cpp:20:10: warning: bad option ‘-favx2’ to attribute ‘optimize’ [-Wattributes]
   20 | int main(){
      |          ^

ソースコード

diff # 

#pragma GCC optimize("avx2")
#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1000000007;
//https://judge.yosupo.jp/submission/5724
long long sum_of_floor_linear(long long a, long long b, long long c, long long n){
    long long tmp=b/c*n+a/c*n*(n-1)/2;
    if(a%c==0){
        return tmp;
    }
    long long next=(c-b%c+a%c-1)/(a%c);
    if(next>=n){
        return tmp;
    }
    a%=c;
    b=b%c+a*next;
    n-=next;
    return tmp+sum_of_floor_linear(c,n*a-((b+a*(n-1))/c*c-b),a,(b+a*(n-1))/c);
}
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++){
      long long tmp = sum_of_floor_linear(A[i], A[i], B[j], B[j]);
      ans += tmp % MOD;
    }
  }
  ans *= 2;
  ans %= MOD;
  cout << ans << endl;
}
0