#1194573 (C++23) No.3777 Train Seats

• 問題: No.3777 Train Seats
• ユーザー: askr58
• 提出日時: 2026-10-23 21:31:55
• 言語: C++23 (gcc 15.3.0 + boost 1.92.0 + ACL)
• 結果: AC
• 実行時間: 36 ms / 2,000 ms
• コード長: 5,891 bytes
• コンパイル時間: 4,253 ms
• コンパイル使用メモリ: 347,776 KB
• 実行使用メモリ: 9,872 KB
• 最終ジャッジ日時: 2026-10-23 21:32:03
• 合計ジャッジ時間: 6,868 ms
• ジャッジサーバーID （参考情報）: judge4_0 / judge2_0

ファイルパターン	結果
sample	AC * 3
other	AC * 50

ソースコード

// SPDX-License-Identifier: CC0-1.0
//
// [English]
// To the extent possible under law, the author(s) have dedicated all copyright
// and related and neighboring rights to this software to the public domain
// worldwide. This software is distributed without any warranty.
// See <https://creativecommons.org/publicdomain/zero/1.0/> for details.
//
// [日本語]
// 法律上可能な範囲で、著作者はこのソフトウェアに関する著作権および
// 著作隣接権を含むすべての関連する権利を、全世界でパブリックドメインに
// 提供しています。本ソフトウェアは一切の保証なく配布されます。
// 自由にコピー・改変・再配布・商用利用できます（クレジット表記は不要です）。
// 詳細は <https://creativecommons.org/publicdomain/zero/1.0/deed.ja> を参照してください。
// ※ 英語の文面および CC0 1.0 の正式な条文（英語）が優先されます。

#include <iostream>
#include <bitset>
#include <random>
#include <chrono>
#include <iomanip>
#include <set>
#include <map>
#include <queue>
#include <deque>
#include <string>
#include <stack>
#include <ranges>
#include <algorithm>
#include <vector>
using namespace std;
using ll=long long;

#include <atcoder/all>
using mint=atcoder::modint998244353;

ostream& operator<<(ostream& os,const mint& x){
	os<<x.val();
	return os;
}
istream& operator>>(istream& is,mint& x){
	int t;
	is>>t;
	x=t;
	return is;
}

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p);
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p);
template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr);
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr);
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec);
template <typename T>
istream& operator>>(istream& is,vector<T>& vec);

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p){
	os<<p.first<<" "<<p.second;
	return os;
}
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p){
	is>>p.first>>p.second;
	return is;
}

template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr){
	for(int i=0;i<n;i++)os<<arr[i]<<(i+1==n?"":" ");
	return os;
}
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr){
	for(int i=0;i<n;i++)is>>arr[i];
	return is;
}
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec){
	for(int i=0;i<(int)vec.size();i++)os<<vec[i]<<(i+1==(int)vec.size()?"":" ");
	return os;
}
template <typename T>
istream& operator>>(istream& is,vector<T>& vec){
	for(int i=0;i<(int)vec.size();i++)is>>vec[i];
	return is;
}

template<class... Vecs>
void input_vec(Vecs&... vs) {
    const auto n = get<0>(tie(vs...)).size();

    for (size_t i = 0; i < n; ++i)
        ((cin >> vs[i]), ...);
}

template <class... Vecs>
void output_vec(const Vecs&... vs) {
    const auto n = get<0>(tie(vs...)).size();

    for (size_t i = 0; i < n; ++i) {
        bool first = true;
        (((cout << (exchange(first, false) ? "" : " ") << vs[i])), ...);
        cout << endl;
    }
}

template <typename T>
vector<T> make_unique(vector<T> vec){
	ranges::sort(vec);
	vec.erase(unique(vec.begin(),vec.end()),vec.end());
	return vec;
}

vector<int> Iota(int n,int s=0){
	vector<int> res(n);
	iota(res.begin(),res.end(),s);
	return res;
}

vector<ll> Iotall(int n,ll s=0){
	vector<ll> res(n);
	iota(res.begin(),res.end(),s);
	return res;
}
void YESNO(bool f){
	if(f)cout<<"Yes"<<endl;
	else cout<<"No"<<endl;
}

using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using vvvl=vector<vector<vector<ll>>>; 
using vi=vector<int>;
using vvi=vector<vector<int>>;
using vvvi=vector<vector<vector<int>>>;

struct Combination{
	
	Combination(int n){
		init(n);
	}
	Combination(){};
	private:
		int n;
		vector<mint> _fact,_factinv;
		void init(int n){
			this->n=n;
			_fact.resize(n+1,1);_factinv.resize(n+1,1);
			for(int i=0;i<n;i++)_fact[i+1]=_fact[i]*(i+1);
			_factinv[n]=_fact[n].inv();
			for(int i=n-1;i>=0;i--)_factinv[i]=_factinv[i+1]*(i+1);
		}
	public:
		mint fact(int x){
			return _fact[x];
		}
		mint factinv(int x){	
			return _factinv[x];
		}
		mint C(int x,int y){
			assert(0<=x&&x<=n&&0<=y);
			if(x<y)return 0;
			else return _fact[x]*_factinv[y]*_factinv[x-y];
		};
		mint Cinv(int x,int y){
			assert(0<=x&&x<=n&&0<=y);
			if(x<y)return 0;
			else return _factinv[x]*_fact[y]*_fact[x-y];
		}
		mint P(int x,int y){
			assert(0<=x&&x<=n&&0<=y);
			if(x<y)return 0; else return _fact[x]*_factinv[x-y];
		}
		mint Pinv(int x,int y){
			assert(0<=x&&x<=n&&0<=y);
			if(x<y)return 0;
			else return _factinv[x]*_fact[x-y];
		}
		// y/x
		mint frac(int y,int x){
			assert(x!=0);
			if(y==0)return 0;
			else return (_fact[y]*_factinv[y-1]*_factinv[x]*_fact[x-1]);
		}
};

Combination comb;

// \sum_{x=0}^{a-1}\binom{b}{x}
// (a,b)->(a-1,b)
void down_binomsum_a(int& a,int& b,mint& c){
	c-=comb.C(b,a-1);
}
//(a,b)->(a+1,b)
void up_binomsum_a(int& a,int& b,mint& c){
	c+=comb.C(b,a);
}
//(a,b)->(a,b-1)
void down_binomsum_b(int& a,int& b,mint& c){
	if(a!=0) c=(c+comb.C(b-1,a-1))*comb.frac(1,2);
}
//(a,b)->(a,b+1)
void up_binomsum_b(int& a,int& b,mint& c){
	if(a!=0) c=2*c-comb.C(b,a-1);
}
int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	int n;
	cin>>n;
	vector<vector<mint>> dp(n+1,vector<mint>(3));
	dp[0][1]=1;
	for(int i=0;i<n;i++){
		vector<vector<mint>> ndp(n+1,vector<mint>(3));
		for(int j=0;j<=n;j++){
			if(j>0){
				ndp[j][0]+=dp[j-1][1]+dp[j-1][2];
			}
			ndp[j][1]=dp[j][0];
			ndp[j][2]=dp[j][1];
		}
		dp=ndp;
	}
	comb=Combination(n+3);
	mint ans=0;
	for(int i=0;i<=n;i++){
		for(int j=0;j<2;j++)ans+=comb.fact(i)*comb.fact(n-i)*dp[i][j];
	}
	cout<<ans<<endl;
}