General
 
 
# Author Problem Lang Verdict Time Memory Sent Judged  
252542234 Practice:
DisconnectedGraph
1799G - 15 C++17 (GCC 7-32) Accepted 15 ms 352 KB 2024-03-21 09:33:55 2024-03-21 09:33:55
→ Source
#include <bits/stdc++.h>
using namespace std;
#define _rep(i_,a_,b_) for(int i_ = (a_); i_ <= (b_); ++i_)
#define mid ((L+R) >> 1)
#define multiCase() int testCnt = in(); _rep(curCase,1,testCnt)
#ifdef ONLINE_JUDGE
#define debug(...) 0
#else
#define debug(...) fprintf(stderr, __VA_ARGS__), fflush(stderr)
#endif
using ll = long long;
using pii = pair<int,int>;

int in(void) { int x; scanf("%d", &x); return x; } ll inl(void) { ll x; scanf("%lld", &x); return x; }
void out(int x) { printf("%d ", x); } void outln(int x) { printf("%d\n", x); }
void out(ll x) { printf("%lld ", x); } void outln(ll x) { printf("%lld\n", x); }
template<typename T> void chkmax(T &a, const T &b) { a = max(a, b); } 
template<typename T> void chkmin(T &a, const T &b) { a = min(a, b); } 
const int p = 998244353, kN = 222;
int n = in(), fac[kN], iv[kN], ifac[kN];
int c[kN]; vector<int> group[kN];
int f[kN][kN], g[kN];
int main() { 
	fac[0] = 1; _rep(i,1,n) fac[i] = 1ll * fac[i - 1] * i % p;
	iv[1] = 1; _rep(i,2,n) iv[i] = 1ll * iv[p % i] * (p - p / i) % p;
	ifac[0] = 1; _rep(i,1,n) ifac[i] = 1ll * ifac[i - 1] * iv[i] % p;
	auto C = [](int n, int m) -> int { return 1ll * fac[n] * ifac[m] % p * ifac[n - m] % p; };
	_rep(i,1,n) c[i] = in();
	_rep(i,1,n) group[in()].emplace_back(i);
	f[0][0] = 1;
	_rep(gid,1,n) { int m = group[gid].size();
		memset(g, 0, sizeof(g));
		g[0] = 1;
		for(auto &v : group[gid])
			for(int i = m; ~i; --i) {
				int newg = 0;
				_rep(b,0,min(i, c[v]))
					newg = (newg + 1ll * g[i - b] * C(m - (i - b), b) % p * ifac[c[v] - b]) % p;
				g[i] = newg;
			}
		_rep(i,0,n) if(f[gid - 1][i]) _rep(j,0,m)
			f[gid][i + j] = (f[gid][i + j] + 1ll * f[gid - 1][i] * g[j]) % p;
	}
	int ans = 0;
	_rep(i,0,n) {
		if((i & 1) == 0) ans = (ans + 1ll * f[n][i] * fac[n - i]) % p;
		else ans = (ans - 1ll * f[n][i] * fac[n - i] % p + p) % p;
	}
	outln(ans);
	return 0;
}

/* 
a list of keywords
clear empty push_back pop_back push pop top front back
emplace_back emplace push_front pop_front insert erase
find count set reset bitset map vector string multiset
first second iterator prev next deque multimap reverse
sort begin end list modify query init check calc prime
putchar getchar puts scanf printf max min swap replace
make_pair make_tuple numeric_limits auto function null
*/
?
Time: ? ms, memory: ? KB
Verdict: ?
Input
?
Participant's output
?
Jury's answer
?
Checker comment
?
Diagnostics
?
Click to see test details