#923257 (C++14) No.2520 L1 Explosion

• 問題: No.2520 L1 Explosion
• ユーザー: kotatsugame
• 提出日時: 2023-10-27 22:07:45
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 29 ms / 2,000 ms
• コード長: 1,388 bytes
• コンパイル時間: 981 ms
• コンパイル使用メモリ: 87,924 KB
• 実行使用メモリ: 4,348 KB
• 最終ジャッジ日時: 2023-10-27 22:07:51
• 合計ジャッジ時間: 2,190 ms
• ジャッジサーバーID （参考情報）: judge13 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,348 KB
testcase_01	AC	2 ms; 4,348 KB
testcase_02	AC	2 ms; 4,348 KB
testcase_03	AC	2 ms; 4,348 KB
testcase_04	AC	1 ms; 4,348 KB
testcase_05	AC	2 ms; 4,348 KB
testcase_06	AC	26 ms; 4,348 KB
testcase_07	AC	9 ms; 4,348 KB
testcase_08	AC	27 ms; 4,348 KB
testcase_09	AC	29 ms; 4,348 KB
testcase_10	AC	27 ms; 4,348 KB
testcase_11	AC	28 ms; 4,348 KB
testcase_12	AC	27 ms; 4,348 KB
testcase_13	AC	26 ms; 4,348 KB
testcase_14	AC	27 ms; 4,348 KB
testcase_15	AC	3 ms; 4,348 KB
testcase_16	AC	19 ms; 4,348 KB
testcase_17	AC	13 ms; 4,348 KB
testcase_18	AC	14 ms; 4,348 KB
testcase_19	AC	12 ms; 4,348 KB
testcase_20	AC	2 ms; 4,348 KB
testcase_21	AC	18 ms; 4,348 KB
testcase_22	AC	4 ms; 4,348 KB
testcase_23	AC	3 ms; 4,348 KB
testcase_24	AC	15 ms; 4,348 KB

ソースコード

#include<iostream>
#include<vector>
#include<algorithm>
#include<cassert>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
int N,M;
int add[3000],cnt[3000];
mint ans[1501];
mint Yd[3000];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N>>M;
	vector<long>Ys;
	Ys.reserve(2*N);
	vector<pair<long,pair<long,bool> > >Xs;
	Xs.reserve(4*N);
	for(int i=0;i<N;i++)
	{
		int X,Y,H;
		cin>>X>>Y>>H;
		long x=X+Y,y=X-Y;
		long d=M-H;
		Ys.push_back(y-d);
		Ys.push_back(y+d);
		Xs.push_back(make_pair(x-d,make_pair(y-d,true)));
		Xs.push_back(make_pair(x-d,make_pair(y+d,false)));
		Xs.push_back(make_pair(x+d,make_pair(y-d,false)));
		Xs.push_back(make_pair(x+d,make_pair(y+d,true)));
	}
	sort(Ys.begin(),Ys.end());
	Ys.erase(unique(Ys.begin(),Ys.end()),Ys.end());
	for(int j=0;j+1<Ys.size();j++)Yd[j]=Ys[j+1]-Ys[j];
	sort(Xs.begin(),Xs.end());
	for(int i=0;;)
	{
		long nowX=Xs[i].first;
		for(int k=0;k<Ys.size();k++)add[k]=0;
		while(i<Xs.size()&&Xs[i].first==nowX)
		{
			int y=lower_bound(Ys.begin(),Ys.end(),Xs[i].second.first)-Ys.begin();
			add[y]+=Xs[i].second.second?1:-1;
			i++;
		}
		if(i==Xs.size())break;
		mint dx=Xs[i].first-nowX;
		for(int k=0;k+1<Ys.size();k++)
		{
			add[k+1]+=add[k];
			cnt[k]+=add[k];
			ans[cnt[k]]+=dx*Yd[k];
		}
	}
	const mint inv2=mint(1)/2;
	for(int i=1;i<=N;i++)cout<<(ans[i]*inv2).val()<<"\n";
}