#15714 (C++11) No.165 四角で囲え！

• 問題: No.165 四角で囲え！
• ユーザー: tkzw_21
• 提出日時: 2015-03-16 14:14:43
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 538 ms / 5,000 ms
• コード長: 1,642 bytes
• コンパイル時間: 1,997 ms
• コンパイル使用メモリ: 167,024 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-12-26 03:24:33
• 合計ジャッジ時間: 6,425 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	160 ms; 5,248 KB
testcase_01	AC	3 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	234 ms; 5,248 KB
testcase_06	AC	62 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	96 ms; 5,248 KB
testcase_12	AC	42 ms; 5,248 KB
testcase_13	AC	42 ms; 5,248 KB
testcase_14	AC	29 ms; 5,248 KB
testcase_15	AC	27 ms; 5,248 KB
testcase_16	AC	334 ms; 5,248 KB
testcase_17	AC	244 ms; 5,248 KB
testcase_18	AC	538 ms; 5,248 KB
testcase_19	AC	286 ms; 5,248 KB
testcase_20	AC	255 ms; 5,248 KB
testcase_21	AC	256 ms; 5,248 KB
testcase_22	AC	256 ms; 5,248 KB

ソースコード

#include <bits/stdc++.h>

using namespace std;

typedef pair<int,int> P;
typedef pair<int,pair<int,int>> PP;
typedef long long ll;

const double EPS = 1e-9;
const int INF = 1e9;
const int MOD = 1e9+7;

int dy[] = {0,1,0,-1};
int dx[] = {1,0,-1,0};

int compress(vector<int>& X){
	vector<int> x = X;
	sort(x.begin(),x.end());
	x.erase(unique(x.begin(),x.end()),x.end());
	for(int i=0;i<X.size();i++){
		X[i] = lower_bound(x.begin(),x.end(),X[i]) - x.begin();
	}
	return x.size();
}
int calc(int l, int r, int u, int d, vector<vector<int> >& sum) {
	if(l-1<0 && u-1<0)return sum[d][r];
	if(l-1<0)return sum[d][r] - sum[u-1][r];
	if(u-1<0)return sum[d][r] - sum[d][l-1];
	return sum[d][r] - sum[d][l-1] - sum[u-1][r] + sum[u-1][l-1];
}

int main(void) {
	int N,B;
	cin >> N >> B;
	vector<int> x(N),y(N),p(N);
	for(int i=0;i<N;i++){
		cin >> x[i] >> y[i] >> p[i];
	}
	//座標圧縮
	int w = compress(x),h = compress(y);

	//２次元配列に代入
	vector<vector<int>> v(h,vector<int>(w)),t(h,vector<int>(w));
	for(int i=0;i<N;i++){
		v[y[i]][x[i]] += p[i];
		t[y[i]][x[i]]++;
	}
	//累積和
	for(int i=0;i<h;i++){
		for(int j=0;j<w-1;j++){
			v[i][j+1] += v[i][j];
			t[i][j+1] += t[i][j];
		}
	}
	for(int j=0;j<w;j++){
		for(int i=0;i<h-1;i++){
			v[i+1][j] += v[i][j];
			t[i+1][j] += t[i][j];
		}
	}

	//x1~x2について全探索
	//その中でy1~y2でしゃくとり法
	int ret = 0;
	for(int x1=0;x1<w;x1++){
		for(int x2=x1;x2<w;x2++){
			int y1=0,y2=0;
			while(y2 < h){
				if(calc(x1,x2,y1,y2,v) <= B){
					ret = max(ret,calc(x1,x2,y1,y2,t));
					y2++;
				}else{
					y1++;
				}
			}
		}
	}
	cout << ret << endl;
	return 0;
}