結果
問題 | No.1043 直列大学 |
ユーザー |
|
提出日時 | 2021-01-17 23:35:10 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,587 bytes |
コンパイル時間 | 129 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 540,544 KB |
最終ジャッジ日時 | 2024-11-30 05:46:25 |
合計ジャッジ時間 | 28,336 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 TLE * 3 MLE * 1 |
ソースコード
#region Header#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsfrom queue import PriorityQueueimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]sys.setrecursionlimit(1000000)#endregion# _INPUT = """3 3# 2 3 4# 10 23 26# 1 100# """# sys.stdin = io.StringIO(_INPUT)MOD = 1000000007def main():N, M = map(int, input().split())V = list(map(int, input().split()))R = list(map(int, input().split()))A, B = map(int, input().split())SumV = sum(V)dpV = [[0 for _ in range(SumV+1)] for _ in range(N)]dpV[0][0] = 1dpV[0][V[0]] = 1for i in range(0, N-1):for v in range(SumV+1):if dpV[i][v] > 0:dpV[i+1][v] += dpV[i][v]dpV[i+1][v+V[i+1]] += dpV[i][v]SumR = sum(R)dpR = [[0 for _ in range(SumR+1)] for _ in range(M)]dpR[0][0] = 1dpR[0][R[0]] = 1for i in range(0, M-1):for r in range(SumR+1):if dpR[i][r] > 0:dpR[i+1][r] += dpR[i][r]dpR[i+1][r+R[i+1]] += dpR[i][r]SV = [0 for _ in range(SumV+1)]SV[0] = dpV[N-1][0]for v in range(1, SumV+1):SV[v] = SV[v-1] + dpV[N-1][v]n = 0for r in range(1, SumR+1):v1 = max(r*A, 1)v2 = min(r*B, SumV)if v1 <= v2:n += dpR[M-1][r] * (SV[v2] - SV[v1-1])print(n % MOD)if __name__ == '__main__':main()