#906971 (C++17) No.2434 RAKUTAN de RAKUTAN

• 問題: No.2434 RAKUTAN de RAKUTAN
• ユーザー: cho435
• 提出日時: 2023-08-22 22:21:19
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,619 bytes
• コンパイル時間: 4,697 ms
• コンパイル使用メモリ: 276,740 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-10 03:56:22
• 合計ジャッジ時間: 5,852 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

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

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)
using mint = atcoder::modint998244353;

int main(){
	int n,h,x;
	cin>>n>>h>>x;
	vector<pair<int,int>> eve;
	int g;
	cin>>g;
	rep(i,g){
		int a;
		cin>>a;
		eve.push_back({a,1});
	}
	int b;
	cin>>b;
	rep(i,b){
		int a;
		cin>>a;
		eve.push_back({a,-1});
	}
	sort(eve.begin(),eve.end());
	if(h>b){
		map<int,int> mp;
		mp[0]=0;
		int bf=0;
		for(auto[nw,dx]:eve){
			auto itr=mp.end();
			itr--;
			mp[nw]=itr->second+dx;
			if(nw<x) continue;
			auto ritr=mp.upper_bound(nw-x);
			auto litr=mp.upper_bound(bf-x);
			if(litr==ritr&&litr!=mp.begin()) litr--;
			while(litr!=ritr){
				mp.at(nw)=max(mp.at(nw),litr->second+dx);
				litr++;
			}
			bf=nw;
		}
		cout<<mp.rbegin()->second<<endl;
		return 0;
	}
	int m=eve.size();
	vector<vector<int>> dp(m+1,vector<int>(h+1,-1e9));
	dp.at(0).at(0)=0;
	int bf=0;
	rep(i,m){
		int l=-1,r=-1;
		if(eve.at(i).first>=x){
			auto ritr=upper_bound(eve.begin(),eve.end(),make_pair(eve.at(i).first-x,2));
			auto litr=upper_bound(eve.begin(),eve.end(),make_pair(bf-x,2));
			if(litr==ritr&&litr!=eve.begin()) litr--;
			l=litr-eve.begin()+1;
			r=ritr-eve.begin()+1;
		}
		rep(j,h+1){
			if(dp.at(i).at(j)!=-1e9) dp.at(i+1).at(j)=max(dp.at(i+1).at(j),dp.at(i).at(j)+eve.at(i).second);
			for(int k=l;k<r;k++){
				if(j<h&&dp.at(k).at(j)!=-1e9){
					dp.at(i+1).at(j+1)=max(dp.at(i+1).at(j+1),dp.at(k).at(j)+eve.at(i).second);
				}
			}
		}
		bf=eve.at(i).first;
	}
	int ans=-1e9;
	rep(j,h+1){
		ans=max(ans,dp.at(m).at(j));
	}
	cout<<ans<<endl;
}