結果

問題 No.1043 直列大学
ユーザー 👑 CleyL
提出日時 2022-05-13 13:33:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 1,288 bytes
コンパイル時間 1,018 ms
コンパイル使用メモリ 76,956 KB
最終ジャッジ日時 2025-01-29 06:26:44
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
using namespace std;
const long long MOD = 1000000007;
int main(){
  int n,m;cin>>n>>m;
  vector<long long> A(n),B(m);
  for(int i = 0; n > i; i++){
    cin>>A[i];
  }
  for(int i = 0; m > i; i++){
    cin>>B[i];
  }
  long long a,b;cin>>a>>b;
  vector<vector<long long>> dp1(2,vector<long long>(101200)),dp2(2,vector<long long>(101200));
  dp1[0][0] = dp2[0][0] = 1;
  for(int i = 0; n > i; i++){
    for(int j = 0; 100000 >= j; j++){
      dp1[1][j+A[i]] = (dp1[1][j+A[i]]+dp1[0][j])%MOD;
      dp1[1][j] = (dp1[1][j]+dp1[0][j])%MOD;
    }
    for(int j = 0; 100000 >= j; j++){
      dp1[0][j] = dp1[1][j];
      dp1[1][j] = 0;
    }
  }
  for(int i = 0; m > i; i++){
    for(int j = 0; 100000 >= j; j++){
      dp2[1][j+B[i]] = (dp2[1][j+B[i]]+dp2[0][j])%MOD;
      dp2[1][j] = (dp2[1][j]+dp2[0][j])%MOD;
    }
    for(int j = 0; 100000 >= j; j++){
      dp2[0][j] = dp2[1][j];
      dp2[1][j] = 0;
    }
  }
  vector<long long> rudp2(101200);
  for(int i = 0; 100000 >= i; i++){
    rudp2[i+1] =  (rudp2[i+1]+rudp2[i]+dp2[0][i])%MOD;
  }
  long long ans = 0;
  for(int i = 1; 100000 >= i; i++){
    long long ru = rudp2[i/a+1] - rudp2[(i+b-1)/b];
    if(ru < 0)ru += MOD;
    ans = (ans+(ru*dp1[0][i])%MOD)%MOD;
  }
  cout << ans << endl;
  
  
}
0