#678892 (C++17) No.1608 Yet Another Ants Problem

• 問題: No.1608 Yet Another Ants Problem
• ユーザー: 沙耶花
• 提出日時: 2021-07-16 22:11:09
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 60 ms / 2,000 ms
• コード長: 1,525 bytes
• コンパイル時間: 4,633 ms
• コンパイル使用メモリ: 268,920 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-07-06 09:28:32
• 合計ジャッジ時間: 6,725 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	4 ms; 6,812 KB
testcase_01	AC	4 ms; 6,940 KB
testcase_02	AC	4 ms; 6,940 KB
testcase_03	AC	4 ms; 6,940 KB
testcase_04	AC	4 ms; 6,940 KB
testcase_05	AC	4 ms; 6,940 KB
testcase_06	AC	4 ms; 6,944 KB
testcase_07	AC	4 ms; 6,940 KB
testcase_08	AC	4 ms; 6,940 KB
testcase_09	AC	4 ms; 6,940 KB
testcase_10	AC	4 ms; 6,940 KB
testcase_11	AC	4 ms; 6,940 KB
testcase_12	AC	4 ms; 6,940 KB
testcase_13	AC	4 ms; 6,944 KB
testcase_14	AC	60 ms; 6,940 KB
testcase_15	AC	59 ms; 6,940 KB
testcase_16	AC	60 ms; 6,940 KB
testcase_17	AC	59 ms; 6,944 KB
testcase_18	AC	60 ms; 6,944 KB
testcase_19	AC	9 ms; 6,940 KB
testcase_20	AC	28 ms; 6,940 KB
testcase_21	AC	28 ms; 6,940 KB
testcase_22	AC	20 ms; 6,940 KB
testcase_23	AC	57 ms; 6,940 KB
testcase_24	AC	59 ms; 6,940 KB
testcase_25	AC	52 ms; 6,944 KB
testcase_26	AC	51 ms; 6,940 KB
testcase_27	AC	52 ms; 6,940 KB
testcase_28	AC	52 ms; 6,940 KB
testcase_29	AC	51 ms; 6,944 KB
testcase_30	AC	60 ms; 6,944 KB
testcase_31	AC	59 ms; 6,940 KB
testcase_32	AC	58 ms; 6,940 KB
testcase_33	AC	58 ms; 6,944 KB
testcase_34	AC	58 ms; 6,944 KB

ソースコード

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000001
struct combi{
	deque<mint> kaijou;
	deque<mint> kaijou_;
	
	combi(int n){
		kaijou.push_back(1);
		for(int i=1;i<=n;i++){
			kaijou.push_back(kaijou[i-1]*i);
		}
		
		mint b=kaijou[n].inv();
		
		kaijou_.push_front(b);
		for(int i=1;i<=n;i++){
			int k=n+1-i;
			kaijou_.push_front(kaijou_[0]*k);
		}
	}
	
	mint combination(int n,int r){
		if(r>n)return 0;
		mint a = kaijou[n]*kaijou_[r];
		a *= kaijou_[n-r];
		return a;
	}
	
	mint junretsu(int a,int b){
		mint x = kaijou_[a]*kaijou_[b];
		x *= kaijou[a+b];
		return x;
	}
	
	mint catalan(int n){
		return combination(2*n,n)/(n+1);
	}
	
};


int main(){
	
	int N,L;
	cin>>N>>L;
	
	vector<int> A(N);
	rep(i,N){
		cin>>A[i];
	}
	
	vector<mint> ans(N,0);
	combi C(100000);
	rep(i,N){
		mint cur = 1;
		for(int j=i+1;j<N;j++){
			if(L - A[j]>A[i]){
				cur = 0;
			}
			else{
			}
		}
		if(cur==0)continue;
		int cnt = 0;
		rep(j,i){
			if(L-A[j]<=A[i])cnt++;
		}
		for(int j=i;j>=0;j--){
			if(cnt>=i-j)ans[j] += C.combination(cnt,i-j);
		}
		
	}
	
	rep(i,N){
		mint cur = 1;
		rep(j,i){
			if(A[j] >= L-A[i]){
				cur = 0;
			}
		}
		if(cur==0)continue;
		int cnt =0;
		for(int j=i+1;j<N;j++){
			if(A[j]<L-A[i])cnt++;
		}
		for(int j=i;j<N;j++){
			if(cnt>=j-i)ans[j] += C.combination(cnt,j-i);
		}
	}
	
	rep(i,N){
		cout<<ans[i].val()<<endl;
	}
	
	return 0;
}