#475474 (C++17) No.1043 直列大学

• 問題: No.1043 直列大学
• ユーザー: hokto
• 提出日時: 2020-05-02 16:49:01
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,086 bytes
• コンパイル時間: 2,587 ms
• コンパイル使用メモリ: 210,768 KB
• 実行使用メモリ: 162,968 KB
• 最終ジャッジ日時: 2024-06-09 17:23:41
• 合計ジャッジ時間: 5,080 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	AC	126 ms; 128,424 KB
testcase_15	AC	147 ms; 152,120 KB
testcase_16	AC	119 ms; 121,336 KB
testcase_17	AC	65 ms; 66,892 KB
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	AC	37 ms; 36,608 KB
testcase_28	AC	110 ms; 109,216 KB
testcase_29	AC	12 ms; 14,192 KB
testcase_30	AC	78 ms; 78,044 KB

ソースコード

#include<bits/stdc++.h>

#define rep(i,n) for(int i=0;i<n;i++)
#define MOD 1000000007
#define MAX_VR 1000
#define ll long long int
using namespace std;
int main()
{
	int N,M;
	cin>>N>>M;
	vector<int> V(N);
	vector<int> R(M);
	sort(V.begin(),V.end());
	sort(R.begin(),R.end());
	rep(i,N) cin>>V[i];
	rep(i,M) cin>>R[i];
	int A,B;
	cin>>A>>B;
	vector<vector<ll>> dp_v(N+1,vector<ll>(N*MAX_VR+1,0));
	vector<vector<ll>> dp_r(M+1,vector<ll>(M*MAX_VR+1,0));
	dp_v[0][0]=1;
	dp_r[0][0]=1;
	rep(i,N)
	{
		for(int j=N*MAX_VR-V[i];j>=0;j--)
		{
			dp_v[i+1][j]+=dp_v[i][j];
			dp_v[i+1][j+V[i]]+=dp_v[i][j];
			dp_v[i+1][j]%=MOD;
		}
	}
	rep(i,M)
	{
		for(int j=M*MAX_VR-R[i];j>=0;j--)
		{
			dp_r[i+1][j]+=dp_r[i][j];
			dp_r[i+1][j+R[i]]+=dp_r[i][j];
			dp_r[i+1][j]%=MOD;
		}
	}
	vector<ll> alloc(N*MAX_VR+2,0);
	rep(i,N*MAX_VR+1)
	{
		alloc[i+1]=alloc[i]+dp_v[N][i];
	}
	ll ans=0;
	for(int i=1;i<=M*MAX_VR;i++)
	{
		int max=B*i;
		int min=A*i;
		if(min>N+MAX_VR)break;
		if(max>N*MAX_VR)max=N*MAX_VR;
		ans+=(alloc[max+1]-alloc[min])*dp_r[M][i];
		ans%=MOD;
	}
	cout<<ans<<endl;
	return 0;
}