結果

問題 No.2434 RAKUTAN de RAKUTAN
ユーザー cho435cho435
提出日時 2023-08-22 22:21:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,619 bytes
コンパイル時間 3,832 ms
コンパイル使用メモリ 264,912 KB
最終ジャッジ日時 2025-02-16 12:20:27
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 9 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
}
0