#905430 (C++14) No.2434 RAKUTAN de RAKUTAN

• 問題: No.2434 RAKUTAN de RAKUTAN
• ユーザー: kotatsugame
• 提出日時: 2023-08-18 21:59:04
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 126 ms / 2,000 ms
• コード長: 1,106 bytes
• コンパイル時間: 915 ms
• コンパイル使用メモリ: 90,456 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-06 04:00:12
• 合計ジャッジ時間: 2,847 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	9 ms; 5,376 KB
testcase_04	AC	9 ms; 5,376 KB
testcase_05	AC	8 ms; 5,376 KB
testcase_06	AC	8 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	124 ms; 5,376 KB
testcase_11	AC	126 ms; 5,376 KB
testcase_12	AC	126 ms; 5,376 KB
testcase_13	AC	125 ms; 5,376 KB
testcase_14	AC	123 ms; 5,376 KB
testcase_15	AC	9 ms; 5,376 KB
testcase_16	AC	9 ms; 5,376 KB
testcase_17	AC	9 ms; 5,376 KB
testcase_18	AC	9 ms; 5,376 KB
testcase_19	AC	9 ms; 5,376 KB
testcase_20	AC	120 ms; 5,376 KB
testcase_21	AC	120 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB
testcase_24	AC	2 ms; 5,376 KB
testcase_25	AC	2 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB

ソースコード

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cassert>
#include<map>
//#include<atcoder/modint>
using namespace std;
//using mint=atcoder::modint998244353;
int N,H,X,G,B;
map<int,int>mp;
int dp[6][5001];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N>>H>>X;
	mp[0]=0;
	mp[N]=0;
	cin>>G;
	for(;G--;)
	{
		int g;cin>>g;
		for(int i=max(g-X,0);i<=min(g+X,N);i++)mp[i]+=0;
		mp[g]+=1;
	}
	cin>>B;
	for(;B--;)
	{
		int g;cin>>g;
		for(int i=max(g-X,0);i<=min(g+X,N);i++)mp[i]+=0;
		mp[g]-=1;
	}
	vector<int>d;
	for(pair<int,int>p:mp)d.push_back(p.second);
	for(int i=0;i<=X;i++)for(int j=0;j<5000;j++)dp[i][j]=-1e9;
	dp[0][0]=0;
	for(int i=0;i+1<d.size();i++)
	{
		int ni=i%(X+1);
		int ti=ni<X?ni+1:0;
		int wi=ni>0?ni-1:X;
		if(i+X>=d.size())wi=-1;
		for(int j=0;j<5000;j++)
		{
			dp[ni][j]+=d[i];
			dp[ti][j]=max(dp[ti][j],dp[ni][j]);
			if(wi>=0)dp[wi][j+1]=max(dp[wi][j+1],dp[ni][j]);
			dp[ni][j]=-1e9;
		}
	}
	int ans=-1e9;
	int ni=(d.size()-1)%(X+1);
	for(int j=0;j<5000&&j<=H;j++)
	{
		ans=max(ans,dp[ni][j]);
	}
	cout<<ans+d.back()<<endl;
}