#999543 (C++14) No.2832 Nana's Fickle Adventure

提出ソース

結果

問題	No.2832 Nana's Fickle Adventure
ユーザー	kotatsugame
提出日時	2024-08-02 23:46:46
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	98 ms / 3,000 ms
コード長	3,943 bytes
コンパイル時間	1,129 ms
コンパイル使用メモリ	79,920 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-08-02 23:46:51
合計ジャッジ時間	4,549 ms
ジャッジサーバーID （参考情報）	judge3 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	24 ms 6,944 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	44 ms 6,944 KB
testcase_04	AC	28 ms 6,940 KB
testcase_05	AC	4 ms 6,940 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	7 ms 6,944 KB
testcase_09	AC	4 ms 6,940 KB
testcase_10	AC	7 ms 6,940 KB
testcase_11	AC	44 ms 6,940 KB
testcase_12	AC	46 ms 6,944 KB
testcase_13	AC	2 ms 6,940 KB
testcase_14	AC	2 ms 6,940 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	2 ms 6,944 KB
testcase_17	AC	2 ms 6,940 KB
testcase_18	AC	98 ms 6,944 KB
testcase_19	AC	89 ms 6,944 KB
testcase_20	AC	98 ms 6,940 KB
testcase_21	AC	94 ms 6,940 KB
testcase_22	AC	75 ms 6,944 KB
testcase_23	AC	98 ms 6,940 KB
testcase_24	AC	88 ms 6,940 KB
testcase_25	AC	88 ms 6,940 KB
testcase_26	AC	96 ms 6,944 KB
testcase_27	AC	4 ms 6,940 KB
testcase_28	AC	6 ms 6,940 KB
testcase_29	AC	48 ms 6,940 KB
testcase_30	AC	69 ms 6,940 KB
testcase_31	AC	84 ms 6,940 KB
testcase_32	AC	25 ms 6,940 KB
testcase_33	AC	2 ms 6,940 KB
testcase_34	AC	2 ms 6,944 KB
testcase_35	AC	7 ms 6,940 KB
testcase_36	AC	2 ms 6,944 KB
testcase_37	AC	85 ms 6,944 KB
testcase_38	AC	3 ms 6,944 KB
testcase_39	AC	46 ms 6,944 KB
testcase_40	AC	23 ms 6,940 KB
testcase_41	AC	46 ms 6,944 KB
testcase_42	AC	48 ms 6,944 KB
testcase_43	AC	89 ms 6,940 KB
testcase_44	AC	3 ms 6,944 KB
testcase_45	AC	3 ms 6,944 KB
testcase_46	AC	3 ms 6,940 KB
testcase_47	AC	2 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<iostream>
#include<cassert>
#include<atcoder/modint>
using namespace std;
#include<vector>
#include<cassert>
template<typename T>
struct Matrix{
	vector<vector<T> >dat;
	int N,M;//N x M matrix
	Matrix(){}
	Matrix(int N_):Matrix(N_,N_){}
	Matrix(int N_,int M_):N(N_),M(M_),dat(N_,vector<T>(M_)){}
	vector<T>&operator[](int i){return dat[i];}
	const vector<T>&operator[](int i)const{return dat[i];}
	static Matrix eye(int N)
	{
		Matrix res(N);
		for(int i=0;i<N;i++)res[i][i]=1;
		return res;
	}
	Matrix operator+(const Matrix&A)const
	{
		assert(N==A.N&&M==A.M);
		Matrix res(N,M);
		for(int i=0;i<N;i++)for(int j=0;j<M;j++)
			res[i][j]=dat[i][j]+A[i][j];
		return res;
	}
	Matrix operator-(const Matrix&A)const
	{
		assert(N==A.N&&M==A.M);
		Matrix res(N,M);
		for(int i=0;i<N;i++)for(int j=0;j<M;j++)
			res[i][j]=dat[i][j]-A[i][j];
		return res;
	}
	Matrix operator*(const Matrix&A)const
	{
		assert(M==A.N);
		Matrix res(N,A.M);
		for(int i=0;i<N;i++)for(int k=0;k<M;k++)for(int j=0;j<A.M;j++)
			res[i][j]+=dat[i][k]*A[k][j];
		return res;
	}
	Matrix pow(long long n)const
	{
		assert(N==M);
		Matrix a=*this,res=eye(N);
		for(;n;a=a*a,n>>=1)if(n&1)res=res*a;
		return res;
	}
	template<typename U>
	Matrix operator+(const U&A)const
	{
		Matrix res(N,M);
		for(int i=0;i<N;i++)for(int j=0;j<M;j++)
			res[i][j]=dat[i][j]+A;
		return res;
	}
	template<typename U>
	Matrix operator-(const U&A)const
	{
		Matrix res(N,M);
		for(int i=0;i<N;i++)for(int j=0;j<M;j++)
			res[i][j]=dat[i][j]-A;
		return res;
	}
	template<typename U>
	Matrix operator*(const U&A)const
	{
		Matrix res(N,M);
		for(int i=0;i<N;i++)for(int j=0;j<M;j++)
			res[i][j]=dat[i][j]*A;
		return res;
	}
	T det()const
	{
		assert(N==M);
		Matrix A=*this;
		T ret=1;
		for(int j=0;j<N;j++)
		{
			int id=-1;
			for(int i=j;i<N;i++)
			{
				if(A[i][j]!=0)
				{
					id=i;
					break;
				}
			}
			if(id==-1)return T(0);
			if(id!=j)
			{
				A[j].swap(A[id]);
				ret=-ret;
			}
			ret*=A[j][j];
			{
				const T a=1/A[j][j];
				for(int k=j+1;k<N;k++)A[j][k]*=a;
			}
			for(int i=j+1;i<N;i++)
			{
				const T a=A[i][j];
				for(int k=j+1;k<N;k++)A[i][k]-=A[j][k]*a;
			}
		}
		return ret;
	}
	int elimination()
	{
		int ret=0;
		for(int j=0;j<M;j++)
		{
			int id=-1;
			for(int i=ret;i<N;i++)
			{
				if(dat[i][j]!=0)
				{
					id=i;
					break;
				}
			}
			if(id==-1)continue;
			if(id!=ret)dat[ret].swap(dat[id]);
			{
				const T a=1/dat[ret][j];
				for(int k=j;k<M;k++)dat[ret][k]*=a;
			}
			for(int i=0;i<N;i++)
			{
				if(i==ret)continue;
				const T a=dat[i][j];
				for(int k=j;k<M;k++)dat[i][k]-=dat[ret][k]*a;
			}
			ret++;
		}
		return ret;
	}
};
using mint=atcoder::modint998244353;
int N,M,X;
int U[1000],V[1000];
int G[10][10];
int deg[10];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N>>M>>X;
	for(int i=0;i<M;i++)
	{
		cin>>U[i]>>V[i];
		U[i]--,V[i]--;
		G[U[i]][V[i]]++;
		deg[U[i]]++;
		if(U[i]!=V[i])
		{
			G[V[i]][U[i]]++;
			deg[V[i]]++;
		}
	}
	if(deg[0]==0)
	{
		cout<<1<<endl;
		for(int i=1;i<N;i++)cout<<0<<endl;
		return 0;
	}
	Matrix<mint>E(N*N,1);
	for(int v=0;v<N;v++)E[v][0]=G[0][v]/mint(deg[0]);
	Matrix<mint>A(N*N);
	for(int v=0;v<N;v++)if(deg[v]>0)
	{
		if(deg[v]==1)
		{
			int u=0;
			while(G[v][u]==0)u++;
			if(u==v)
			{
				A[v*N+v][v*N+v]+=1;
			}
			else
			{
				A[v*N+v][u*N+v]+=1;
				A[v*N+u][v*N+v]+=1;
			}
		}
		else
		{
			for(int u=0;u<N;u++)if(G[u][v]>0)
			{
				int id=u*N+v;
				for(int w=0;w<N;w++)if(G[v][w]>0)
				{
					int jd=v*N+w;
					if(w!=u)
					{
						A[jd][id]+=G[v][w]/mint(deg[v]-1);
					}
					else
					{
						A[jd][id]+=(G[v][w]-1)/mint(deg[v]-1);
					}
				}
			}
		}
	}
	/*
	for(int i=0;i<N+N;i++)
	{
		for(int j=0;j<N+N;j++)cout<<A[i][j].val()<<" ";
		cout<<endl;
	}
	for(int i=0;i<N+N;i++)cout<<E[i][0].val()<<endl;
	*/
	A=A.pow(X-1)*E;
	for(int v=0;v<N;v++)
	{
		mint ans=0;
		for(int u=v;u<N*N;u+=N)ans+=A[u][0];
		cout<<ans.val()<<"\n";
	}
}

yukicoder

結果

テストケース

ソースコード