#789631 (C++17) No.165 四角で囲え！

• 問題: No.165 四角で囲え！
• ユーザー: ttkkggww
• 提出日時: 2022-08-18 20:30:22
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 146 ms / 5,000 ms
• コード長: 1,349 bytes
• コンパイル時間: 4,920 ms
• コンパイル使用メモリ: 281,416 KB
• 実行使用メモリ: 6,820 KB
• 最終ジャッジ日時: 2024-10-06 15:17:12
• 合計ジャッジ時間: 8,125 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	109 ms; 5,248 KB
testcase_01	AC	88 ms; 6,820 KB
testcase_02	AC	87 ms; 6,820 KB
testcase_03	AC	88 ms; 6,820 KB
testcase_04	AC	88 ms; 6,816 KB
testcase_05	AC	131 ms; 6,820 KB
testcase_06	AC	105 ms; 6,816 KB
testcase_07	AC	88 ms; 6,816 KB
testcase_08	AC	87 ms; 6,816 KB
testcase_09	AC	88 ms; 6,820 KB
testcase_10	AC	87 ms; 5,248 KB
testcase_11	AC	104 ms; 5,248 KB
testcase_12	AC	92 ms; 5,248 KB
testcase_13	AC	98 ms; 5,248 KB
testcase_14	AC	95 ms; 5,248 KB
testcase_15	AC	90 ms; 5,248 KB
testcase_16	AC	129 ms; 5,248 KB
testcase_17	AC	88 ms; 5,248 KB
testcase_18	AC	146 ms; 5,248 KB
testcase_19	AC	106 ms; 5,248 KB
testcase_20	AC	88 ms; 5,248 KB
testcase_21	AC	88 ms; 5,248 KB
testcase_22	AC	89 ms; 5,248 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#include<atcoder/all>
using namespace atcoder;
using ll = long long;
int n,b;
vector<int> x,y,p;

void solve(){
	{
		set<int> st;
		for(int i = 0;i<n;i++){
			st.insert(x[i]);
		}
		int idx = 0;
		map<int,int> mp;
		for(auto &i:st){
			mp[i] = idx++;
		}
		for(int i = 0;i<n;i++){
			x[i] = mp[x[i]];
		}
	}
	{
		set<int> st;
		for(int i = 0;i<n;i++){
			st.insert(y[i]);
		}
		int idx = 0;
		map<int,int> mp;
		for(auto &i:st){
			mp[i] = idx++;
		}
		for(int i = 0;i<n;i++){
			y[i] = mp[y[i]];
		}
	}
	vector<vector<pair<int,int>>> binp(400);
	for(int i = 0;i<n;i++){
		//cerr<<x[i]<<' '<<y[i]<<' '<<p[i]<<endl;
		binp[x[i]].push_back(pair<int,int>(y[i],p[i]));
	}
	int ans = 0;
	for(int _ = 0;_<400;_++){
		vector<int> sump(400),sumc(400);
	for(int x =_;x<400;x++){
		for(auto &[y,p]:binp[x]){
			sump[y] += p;
			sumc[y] += 1;
		}
		queue<int> Q;
		int curSum=0;
		int cur  = 0;
		for(int i = 0;i<400;i++){
			Q.push(i);
			cur += sumc[i];
			curSum += sump[i];
			while(curSum>b){
				int idx = Q.front();
				cur -= sumc[idx];
				curSum -= sump[idx];
				Q.pop();
			}
			ans = max(ans,cur);
		}
	}
	}
	cout<<ans<<endl;
}

signed main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cin >> n >> b;
	x = y = p = vector<int>(n);
	for(int i = 0;i<n;i++)cin >> x[i] >> y[i] >> p[i];
	solve();
}