結果

問題 No.165 四角で囲え！
ユーザー koyumeishikoyumeishi
提出日時 2015-03-13 00:29:36
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,515 ms / 5,000 ms
コード長 2,142 bytes
コンパイル時間 978 ms
コンパイル使用メモリ 83,920 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-07 18:58:32
合計ジャッジ時間 15,585 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 726 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 917 ms
5,376 KB
testcase_06 AC 136 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 374 ms
5,376 KB
testcase_12 AC 154 ms
5,376 KB
testcase_13 AC 144 ms
5,376 KB
testcase_14 AC 91 ms
5,376 KB
testcase_15 AC 108 ms
5,376 KB
testcase_16 AC 1,420 ms
5,376 KB
testcase_17 AC 1,483 ms
5,376 KB
testcase_18 AC 1,311 ms
5,376 KB
testcase_19 AC 1,508 ms
5,376 KB
testcase_20 AC 1,515 ms
5,376 KB
testcase_21 AC 1,500 ms
5,376 KB
testcase_22 AC 1,504 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;

int main(){
	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];

	auto compress = [&](vector<int>& V){
		auto V__ = V;
		sort(V__.begin(), V__.end());
		V__.erase( unique(V__.begin(), V__.end()), V__.end());

		for(int i=0; i<N; i++){
			V[i] = lower_bound(V__.begin(), V__.end(), V[i]) - V__.begin();
		}
	};

	compress(X); compress(Y);

	int width = *max_element(X.begin(), X.end()) +1;
	int height = *max_element(Y.begin(), Y.end()) +1;

	vector<vector<int>> sum_p(width, vector<int>(height, 0));
	vector<vector<int>> sum_c(width, vector<int>(height, 0));
	for(int i=0; i<N; i++){
		sum_p[ X[i] ][ Y[i] ] += P[i];
		sum_c[ X[i] ][ Y[i] ] += 1;
	}

	for(int i=0; i<width; i++){
		for(int j=0; j<height; j++){
			if(j>0){
				sum_p[i][j] += sum_p[i][j-1];
				sum_c[i][j] += sum_c[i][j-1];
			}
			if(i>0){
				sum_p[i][j] += sum_p[i-1][j];
				sum_c[i][j] += sum_c[i-1][j];
			}
			if(i>0 && j>0){
				sum_p[i][j] -= sum_p[i-1][j-1];
				sum_c[i][j] -= sum_c[i-1][j-1];
			}
		}
	}

	//[ (a,b) , (c,d) ]
	auto get_points = [&](int a, int b, int c, int d){
		int ret = 0;
		ret += sum_p[c][d];
		if(a>0) ret -= sum_p[a-1][d];
		if(b>0) ret -= sum_p[c][b-1];
		if(a>0 && b>0) ret += sum_p[a-1][b-1];

		return ret;
	};


	//[ (a,b) , (c,d) ]
	auto get_counts = [&](int a, int b, int c, int d){
		int ret = 0;
		ret += sum_c[c][d];
		if(a>0) ret -= sum_c[a-1][d];
		if(b>0) ret -= sum_c[c][b-1];
		if(a>0 && b>0) ret += sum_c[a-1][b-1];

		return ret;
	};

	int ans = 0;
	for(int i=0; i<width; i++){
		for(int j=0; j<height; j++){
			for(int k=i; k<width; k++){
				int lb = j;
				int ub = height;

				while(ub-lb>1){
					int med = (lb+ub)/2;
					int val = get_points(i,j, k,med);
					if(val > B){
						ub = med;
					}else{
						lb = med;
					}
				}

				if(get_points(i,j, k,lb) <= B)	ans = max(ans, get_counts(i,j, k,lb));
			}
		}
	}

	cout << ans << endl;


	return 0;
}
0