#465184 (C++17) No.310 2文字しりとり

提出ソース

結果

問題	No.310 2文字しりとり
ユーザー	HIR180
提出日時	2020-04-17 15:37:39
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	614 ms / 6,000 ms
コード長	5,010 bytes
コンパイル時間	3,566 ms
コンパイル使用メモリ	237,304 KB
実行使用メモリ	67,072 KB
最終ジャッジ日時	2024-10-03 10:20:51
合計ジャッジ時間	6,875 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 KB
testcase_03	AC	2 ms 5,248 KB
testcase_04	AC	2 ms 5,248 KB
testcase_05	AC	2 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	2 ms 5,248 KB
testcase_08	AC	2 ms 5,248 KB
testcase_09	AC	2 ms 5,248 KB
testcase_10	AC	2 ms 5,248 KB
testcase_11	AC	2 ms 5,248 KB
testcase_12	AC	2 ms 5,248 KB
testcase_13	AC	2 ms 5,248 KB
testcase_14	AC	2 ms 5,248 KB
testcase_15	AC	2 ms 5,248 KB
testcase_16	AC	1 ms 5,248 KB
testcase_17	AC	2 ms 5,248 KB
testcase_18	AC	3 ms 5,248 KB
testcase_19	AC	4 ms 5,248 KB
testcase_20	AC	5 ms 5,248 KB
testcase_21	AC	496 ms 66,776 KB
testcase_22	AC	591 ms 66,944 KB
testcase_23	AC	614 ms 66,944 KB
testcase_24	AC	585 ms 67,072 KB
testcase_25	AC	593 ms 66,944 KB
testcase_26	AC	124 ms 20,608 KB
testcase_27	AC	168 ms 27,776 KB

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

コンパイルメッセージ

main.cpp: In function 'std::vector<_Tp> BerlekampMassey(std::vector<_Tp>) [with T = int]':
main.cpp:87:44: warning: 'ld' may be used uninitialized [-Wmaybe-uninitialized]
   87 |                 T k = (ll) -(x[i]-t)*modpow(ld,mod-2)%mod;
      |                                      ~~~~~~^~~~~~~~~~
main.cpp:76:14: note: 'ld' was declared here
   76 |         T lf,ld;
      |              ^~
main.cpp:88:29: warning: 'lf' may be used uninitialized [-Wmaybe-uninitialized]
   88 |                 vector<T>c(i-lf-1);
      |                            ~^~~
main.cpp:76:11: note: 'lf' was declared here
   76 |         T lf,ld;
      |           ^~
main.cpp: In function 'int main()':
main.cpp:207:28: warning: 'idi' may be used uninitialized [-Wmaybe-uninitialized]
  207 |                 outdeg[idi]++;
      |                 ~~~~~~~~~~~^~
main.cpp:189:22: note: 'idi' was declared here
  189 |         int bad = 0, idi, ido;
      |                      ^~~

ソースコード

raw source code

//Let's join Kaede Takagaki Fan Club !!
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
typedef pair<int,P> P1;
typedef pair<P,P> P2;
#define pu push
#define pb push_back
#define mp make_pair
#define eps 1e-7
#define INF 1000000000
#define fi first
#define sc second
#define rep(i,x) for(int i=0;i<x;i++)
#define repn(i,x) for(int i=1;i<=x;i++)
#define SORT(x) sort(x.begin(),x.end())
#define ERASE(x) x.erase(unique(x.begin(),x.end()),x.end())
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin())
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin())
#define all(x) x.begin(),x.end()
template<class T>
void dmp(T a){
	rep(i,a.size()) cout << a[i] << " ";
	cout << endl;
}
template<class T>
bool chmax(T&a, T b){
	if(a < b){
		a = b;
		return 1;
	}
	return 0;
}
template<class T>
bool chmin(T&a, T b){
	if(a > b){
		a = b;
		return 1;
	}
	return 0;
}
template<class T>
void g(T &a){
	cin >> a;
}
template<class T>
void o(const T &a,bool space=false){
	cout << a << (space?' ':'\n');
}
//ios::sync_with_stdio(false);
const int mod = 1000000007;
template<class T>
void add(T&a,T b){
	a+=b;
	if(a >= mod) a-=mod;
}
template<class T>
T mul(T a, T b){
	return (T) ((ll)(a) * b % mod);
}
ll modpow(ll x,ll n){
	ll res=1;
	while(n>0){
		if(n&1) res=res*x%mod;
		x=x*x%mod;
		n>>=1;
	}
	return res;
}
template<class T>
vector<T>BerlekampMassey(vector<T>x){
	vector<T>ls,cur;
	T lf,ld;
	rep(i,x.size()){
		T t = 0;
		for(int j=0;j<cur.size();j++){
			t = (t+(ll)x[i-j-1]*cur[j])%mod;
		}
		if( ((t-x[i])%mod+mod)%mod == 0 ) continue;
		if(!cur.size()){
			cur.resize(i+1); lf = i; ld = (t-x[i])%mod;
			continue;
		}
		T k = (ll) -(x[i]-t)*modpow(ld,mod-2)%mod;
		vector<T>c(i-lf-1);
		c.pb(k);
		rep(j,ls.size()) c.pb((ll)-ls[j]*k%mod);
		if(c.size() < cur.size()) c.resize(cur.size());
		rep(j,cur.size()){
			c[j]=(c[j]+cur[j]);
			while(c[j] < 0) c[j] += mod;
			while(c[j] >= mod) c[j] -= mod;
		}
		if(i-lf+(int)(ls.size()) >= (int)(cur.size())){
			ls = cur, lf = i, ld = (t-x[i])%mod;
		}
		cur = c;
	}
	rep(i,cur.size()) cur[i] = (cur[i]%mod+mod)%mod;
	return cur;
}

vector<int>rand_vec(int n, int seed = -1){
	if(seed == -1) seed = n;
	mt19937 mt(seed);
	vector<int>ret(n);
	rep(i, n) ret[i] = mt() % (mod-1) + 1;
	return ret;
}
int dot(vector<int>&a, vector<int>&b){
	int ret = 0;
	rep(i, a.size()) add(ret, mul(a[i], b[i]));
	return ret;
}
//今回はunmarkのcellはM_(i,j) = -v[j]なので注意
vector<int>find_min_poly(int n, vector<P1>mat, vector<int>&v){
	srand((unsigned)time(NULL));
	vector<int>M = rand_vec(n), R = rand_vec(n, rand());
	vector<int>test(2*n);
	rep(i, 2*n){
		test[i] = dot(M, R);
		vector<int>nw(n);
		rep(j, n) add(nw[0], mul(M[j], mod-v[j]));
		rep(i, n-1) nw[i+1] = nw[0];
		
		rep(j, mat.size()){
			int r = mat[j].sc.fi, c = mat[j].sc.sc;
			add(nw[r], mul((v[c]+mat[j].fi), M[c]));
		}
		swap(nw, M);
	}
	vector<int>ret = BerlekampMassey(test);
	for(int i=0;i<ret.size();i++){
		if(ret[i]) ret[i] = mod-ret[i];
	}
	reverse(all(ret));
	ret.pb(1);
	return ret;
}
//今回はunmarkのcellは-1なので注意
//0-indexed
int calc_det(int n, vector<P1>mat){
    if(n <= 0) return 1;
	srand((unsigned)time(NULL));
	while(1){
		vector<int>M = rand_vec(n, rand());
		vector<P1>cp = mat;
		rep(j, mat.size()){
			int r = mat[j].sc.fi, c = mat[j].sc.sc;
			cp[j].fi = mul(cp[j].fi, M[c]);
		}
		vector<int>ret = find_min_poly(n, cp, M);
		if(ret[0] == 0) return 0;
		if(ret.size() != n+1) continue;
		int ans = (n%2 == 0 ? ret[0] : mod-ret[0]);
		int d1 = 1;
		for(auto a:M) d1 = mul(d1, a);
		d1 = modpow(d1, mod-2);
		return (mul(ans, d1)%mod+mod)%mod;
	}
}


int n, m;
int indeg[4005], outdeg[4005];
int cnt[4005][4005];
bool cir;
ll F[4005];

int main(){
	scanf("%d%d", &n, &m);
	if(m == n*n){
	    puts("1"); return 0;
	}
	repn(i, n) repn(j, n){
		cnt[i][j] ++;
		indeg[j] ++;
		outdeg[i] ++;
	}
	rep(i, m){
		int a, b; scanf("%d%d", &a, &b);
		cnt[a][b] --;
		indeg[b] --;
		outdeg[a] --;
	}
	int bad = 0, idi, ido;
	repn(i, n){
		if(outdeg[i]-indeg[i] == 1) {
			bad ++;
			ido = i;
		}
		else if(outdeg[i]-indeg[i] == -1) {
			idi = i;
		}
		else if(indeg[i] != outdeg[i]) bad = 61471;
	}
	if(bad == 0) cir = 1;
	else if(bad > 1){
		puts("0"); return 0;
	}
	else{
		cnt[idi][ido]++;
		indeg[ido]++;
		outdeg[idi]++;
	}
	vector<P1>mat;
	vector<int>conv(n+2), rev(n+2);
	int sz = 0;
	repn(i, n){
	    if(indeg[i]){
	        rev[sz] = i;
	        conv[i] = sz++;
	    }
	}
	rep(i, sz-1) rep(j, sz-1){
		if(i != j){
			if(cnt[rev[i]][rev[j]] != 1) mat.pb(mp(-cnt[rev[i]][rev[j]], mp(i, j)));
		}
		else{
			mat.pb(mp(indeg[rev[i]]-cnt[rev[i]][rev[i]], mp(i, i)));
		}
	}
	int ctree = calc_det(sz-1, mat);
	if(ctree == 0){
		puts("0"); return 0;
	}
	F[0] = 1;
	for(int i=1;i<4005;i++) F[i] = F[i-1] * i % mod;
	
	repn(i,n) if(indeg[i]) ctree = mul(ctree, (int)F[indeg[i]-1]);
	if(cir) ctree = mul(ctree, n*n-m);
	
	printf("%d\n", ctree);
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード