#924496 (C++17) No.2520 L1 Explosion

• 問題: No.2520 L1 Explosion
• ユーザー: cho435
• 提出日時: 2023-10-30 21:47:17
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 292 ms / 2,000 ms
• コード長: 1,386 bytes
• コンパイル時間: 4,964 ms
• コンパイル使用メモリ: 272,460 KB
• 実行使用メモリ: 82,816 KB
• 最終ジャッジ日時: 2024-09-25 17:24:27
• 合計ジャッジ時間: 8,375 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	270 ms; 82,688 KB
testcase_07	AC	231 ms; 82,816 KB
testcase_08	AC	292 ms; 82,816 KB
testcase_09	AC	191 ms; 82,688 KB
testcase_10	AC	285 ms; 82,688 KB
testcase_11	AC	286 ms; 82,816 KB
testcase_12	AC	291 ms; 82,816 KB
testcase_13	AC	289 ms; 82,816 KB
testcase_14	AC	286 ms; 82,816 KB
testcase_15	AC	12 ms; 6,400 KB
testcase_16	AC	200 ms; 59,904 KB
testcase_17	AC	129 ms; 39,296 KB
testcase_18	AC	138 ms; 40,832 KB
testcase_19	AC	112 ms; 33,920 KB
testcase_20	AC	5 ms; 5,376 KB
testcase_21	AC	189 ms; 55,936 KB
testcase_22	AC	24 ms; 9,216 KB
testcase_23	AC	14 ms; 6,656 KB
testcase_24	AC	163 ms; 48,000 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(){
	ll n,m;
	cin>>n>>m;
	vector<ll> x(n),y(n),h(n);
	rep(i,n) cin>>x.at(i)>>y.at(i)>>h.at(i);
	vector<ll> a(n),b(n);
	rep(i,n){
		a.at(i)=x.at(i)+y.at(i);
		b.at(i)=x.at(i)-y.at(i);
	}
	auto ca=a,cb=b;
	rep(i,n){
		ca.push_back(a.at(i)-(m-h.at(i)));
		ca.push_back(a.at(i)+(m-h.at(i)));
		cb.push_back(b.at(i)-(m-h.at(i)));
		cb.push_back(b.at(i)+(m-h.at(i)));
	}
	sort(ca.begin(),ca.end());
	sort(cb.begin(),cb.end());
	int l=ca.size();
	vector<vector<int>> vn(l+1,vector<int>(l+1,0));
	rep(i,n){
		int la=lower_bound(ca.begin(),ca.end(),a.at(i)-(m-h.at(i)))-ca.begin();
		int ua=lower_bound(ca.begin(),ca.end(),a.at(i)+(m-h.at(i)))-ca.begin();
		int lb=lower_bound(cb.begin(),cb.end(),b.at(i)-(m-h.at(i)))-cb.begin();
		int ub=lower_bound(cb.begin(),cb.end(),b.at(i)+(m-h.at(i)))-cb.begin();
		vn.at(la).at(lb)++;
		vn.at(la).at(ub)--;
		vn.at(ua).at(lb)--;
		vn.at(ua).at(ub)++;
	}
	rep(i,l+1) rep(j,l) vn.at(i).at(j+1)+=vn.at(i).at(j);
	rep(i,l) rep(j,l+1) vn.at(i+1).at(j)+=vn.at(i).at(j);
	vector<mint> ans(n);
	rep(i,l){
		rep(j,l){
			if(vn.at(i).at(j)==0) continue;
			ans.at(vn.at(i).at(j)-1)+=(ca.at(i+1)-ca.at(i))*(cb.at(j+1)-cb.at(j));
		}
	}
	rep(i,n) cout<<(ans.at(i)/2).val()<<endl;
}