QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#708794#4781. 完美的集合NineSuns19 37ms344200kbC++144.9kb2024-11-04 06:44:142024-11-04 06:44:14

Judging History

你现在查看的是最新测评结果

  • [2024-11-04 06:44:14]
  • 评测
  • 测评结果:19
  • 用时:37ms
  • 内存:344200kb
  • [2024-11-04 06:44:14]
  • 提交

answer

#include <bits/stdc++.h>
#define ll long long
#define pii pair <int, int>
#define fi first
#define se second
#define pb push_back
#define pil pair <int, ll> 
#define LL __int128

using namespace std;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
const int N = 65, S = 10005; 
const ll mod = 11920928955078125, inf = 0x3f3f3f3f3f3f3f3f, imod = mod/5*4; 
int n, m, c, mk[N][N], vis[N], sm[N], sk[N]; 
ll M, w[N], v[N], k, dis[N][N], f[N][N][S], g[S]; 
vector <pil> e[N];

void dfs (int rt, int p, int fa) {
	for (auto i : e[p]) {
		if (i.fi == fa) continue; 
		dis[rt][i.fi] = dis[rt][p]+i.se; dfs(rt, i.fi, p); 
	}
}

struct val {
	ll v, c; 
	friend val operator * (const val & a, const val & b) {
		return (val){a.v+b.v, a.c*b.c}; 
	} 
	friend val operator + (const val & a, const val & b) {
		if (a.v^b.v) return a.v > b.v ? a : b; 
		return (val){a.v, a.c+b.c};  
	}
	void clr () {
		v = -inf; c = 0; 
	}
}fs[N][S], gs[S]; 

vector <int> td; 
void dp (int p, int fa) {
	if (mk[p][fa]) return; 
	mk[p][fa] = 1; 
	if (w[p] <= m) f[p][fa][w[p]] = v[p]; 
	for (auto i : e[p]) {
		dp(i.fi, p); 
		memset(g, -0x3f, sizeof g);
		td.clear(); 
		for (int j = 0;j <= m;j++) if (f[i.fi][p][j] >= 0) td.pb(j); 
		for (int j = 0;j <= m;j++) {
			if (f[p][fa][j] < 0) continue; 
			for (int z : td) if (j+z <= m) g[j+z] = max(g[j+z], f[i.fi][p][z]+f[p][fa][j]); 
		} 
		memcpy(f[p][fa], g, sizeof g); 
	}
	f[p][fa][0] = 0; 
}
inline bool chk (int i, int j) {
	return v[j] == 0 || dis[i][j] <= M/v[j]; 
}

void dp1 (int p, int fa) {
	for (int j = 0;j <= m;j++) fs[p][j].clr();
	if (vis[p]) {
		if (w[p] <= m) fs[p][w[p]] = (val){v[p], 1}; 
//		cout << "FFFFFFFFFFF" << " " << p << " ADDDDDDDDDDDDDDDD " << "FEFJK" << endl; 
//		for (int j = 0;j <= m;j++) cout << j << " " << fs[p][j].v << " " << fs[p][j].c << endl; 
		for (auto i : e[p]) {
			if (i.fi == fa) continue; 
			dp1(i.fi, p);
			td.clear(); 
			for (int j = 0;j <= m;j++) gs[j].clr(); 
			for (int j = 0;j <= m;j++) if (fs[i.fi][j].v >= 0) td.pb(j); 
			for (int j = 0;j <= m;j++) {
				if (fs[p][j].v < 0) continue;
				for (int z : td) if (j+z <= m) {
//					if (j+z == 1 && p == 1 && i.fi == 2) cout << gs[j+z].v << " " << gs[j+z].c << " " << fs[p][j].c << " CHKLKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK " << fs[i.fi][z].c << endl; 
					gs[j+z] = gs[j+z]+(fs[p][j]*fs[i.fi][z]);  
					
				}
			}
			memcpy(fs[p], gs, sizeof gs); 
//		cout << "FFFFFFFFFFF" << " " << p << " ADDDDDDDDDDDDDDDD " << i.fi << endl; 
//		for (int j = 0;j <= m;j++) cout << j << " " << fs[p][j].v << " " << fs[p][j].c << endl; 
//		cout << "END: F\n"; 
		}
	}
	if (!sk[p]) fs[p][0] = (val){0, 1}; 
}

ll qpow (LL x, ll y = imod-1) {
	LL res = 1;
	while (y) {
		if (y&1) res = res*x%mod;
		x = x*x%mod; y >>= 1;
	}
	return res; 
}

ll C (ll n, ll m) {
	if (m > n) return 0; 
	if (m > n-m) m = n-m; 
	LL res = 1; 
	ll c = 0, x = 1; 
	LL k = n; while (k) c += k/5, k /= 5; 
	k = m; while (k) c -= k/5, k /= 5; 
	k = n-m; while (k) c -= k/5, k /= 5;
	if (c > 22) return 0; 
	while (c--) x = x*5%mod; 
	for (ll i = n;i > n-m;i--) {
		k = i; while (k%5 == 0) k /= 5; 
		res = res*k%mod; 
	}
	for (ll i = 1;i <= m;i++) {
		k = i; while (k%5 == 0) k /= 5; 
		res = res*qpow(k)%mod; 
	}
	return res*x%mod; 
}

void solve () {
	memset(f, -0x3f, sizeof f); 
	cin >> n >> m >> k >> M;
	for (int i = 1;i <= n;i++) cin >> w[i];
	for (int i = 1;i <= n;i++) cin >> v[i]; 
	for (int i = 1;i < n;i++) {
		int x, y; ll c; cin >> x >> y >> c; 
		e[x].pb({y, c}); e[y].pb({x, c});
	}
	for (int i = 1;i <= n;i++) dfs(i, i, 0); 
	for (int i = 1;i <= n;i++) dp(i, 0); 
	ll ans = -inf, cnt = 0; 
	for (int i = 1;i <= n;i++) for (int j = 0;j <= m;j++) ans = max(ans, f[i][0][j]); 
	for (int i = 1;i <= n;i++) {
		memset(sk, 0, sizeof sk); 
		for (int j = 1;j <= n;j++) vis[j] = chk(i, j); 
		sk[i] = 1; dp1(i, 0); 
		ll sc = 0; 
		for (int j = 0;j <= m;j++) if (fs[i][j].v == ans) sc += fs[i][j].c; 
		cnt += C(sc, k), cnt %= mod; 
	}
	for (int i = 1;i <= n;i++) {
		for (auto j : e[i]) {
			if (j.fi > i) continue; 
			memset(sk, 0, sizeof sk); 
			for (int z = 1;z <= n;z++) vis[z] = chk(i, z) && chk(j.fi, z); 
			sk[i] = sk[j.fi] = 1; 
			dp1(i, j.fi); dp1(j.fi, i); 
			
			for (int z = 0;z <= m;z++) gs[z].clr(); 
			td.clear(); 
			for (int z = 0;z <= m;z++) if (fs[j.fi][z].v >= 0) td.pb(z); 
			for (int z = 0;z <= m;z++) {
				if (fs[i][z].v < 0) continue;
				for (int x : td) if (z+x <= m) gs[z+x] = gs[z+x]+(fs[i][z]*fs[j.fi][x]);  
			}
			ll sc = 0; 
			for (int z = 0;z <= m;z++) if (gs[z].v == ans) sc += gs[z].c; 
//			cout << i << " " << j.fi << " " << -sc << " " << C(sc, k) << endl; 
			cnt -= C(sc, k), cnt %= mod; 
		}
	}
	(cnt += mod) %= mod; 
	cout << cnt; 
}

signed main () {
//	ios::sync_with_stdio(0);
//	cin.tie(0); cout.tie(0);
	int T = 1; 
	while (T--) solve();
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 13
Accepted
time: 3ms
memory: 340148kb

input:

16 109 1 4025082
46 68 46 1 46 67 111 1 156 1 45 45 1 45 45 45
8525 12789 8526 0 8526 12788 954 0 6 0 8525 8526 0 8525 8526 8526
1 2 290
1 3 188
1 4 420
1 5 6
2 6 29
1 7 643
1 8 461
4 9 468
1 10 228
5 11 428
2 12 71
4 13 290
1 14 957
2 15 955
4 16 549

output:

25

result:

ok 1 number(s): "25"

Test #2:

score: 13
Accepted
time: 11ms
memory: 337876kb

input:

17 144 2 3550388
2 1 3 1 60 88 2 2 59 60 88 2 1 2 1 89 60
0 0 0 0 6962 10443 0 0 6962 6962 10442 0 0 0 0 10443 6962
1 2 25
1 3 715
1 4 337
1 5 267
1 6 146
1 7 634
5 8 208
1 9 562
1 10 134
2 11 984
2 12 891
3 13 330
5 14 854
3 15 961
3 16 679
5 17 388

output:

116886

result:

ok 1 number(s): "116886"

Test #3:

score: 13
Accepted
time: 12ms
memory: 337884kb

input:

17 131 6 4918336
68 67 1 46 134 45 46 1 45 1 45 67 1 1 1 1 45
10569 10568 0 7046 9079 7046 7046 0 7046 0 7045 10569 0 0 0 0 7045
1 2 357
1 3 219
2 4 379
1 5 683
1 6 772
1 7 125
1 8 297
1 9 912
3 10 438
5 11 319
2 12 850
1 13 280
2 14 925
1 15 20
1 16 412
1 17 718

output:

12271512

result:

ok 1 number(s): "12271512"

Test #4:

score: 0
Wrong Answer
time: 15ms
memory: 339812kb

input:

16 51 4 497
7 2 6 8 9 1 5 9 4 6 8 7 4 9 6 3
40 16 48 56 72 8 40 56 16 32 48 56 32 56 32 8
1 2 5
2 3 2
1 4 3
1 5 3
4 6 5
3 7 3
2 8 3
3 9 5
1 10 1
5 11 4
5 12 1
6 13 4
7 14 5
3 15 5
1 16 1

output:

0

result:

wrong answer 1st numbers differ - expected: '1', found: '0'

Subtask #2:

score: 0
Wrong Answer

Test #13:

score: 0
Wrong Answer
time: 19ms
memory: 341928kb

input:

40 816 1 285
46 124 125 137 90 33 15 73 67 41 134 106 3 163 152 151 14 77 157 82 40 9 151 148 60 60 163 71 40 134 152 145 70 59 26 64 94 38 158 57
2 2 1 1 1 1 3 3 1 1 3 3 1 3 1 3 2 1 3 3 1 1 2 2 3 2 1 2 1 2 2 1 3 3 3 1 1 1 3 3
1 2 20
1 3 26
2 4 14
3 5 25
1 6 17
4 7 49
1 8 37
1 9 50
2 10 52
4 11 55
2...

output:

0

result:

wrong answer 1st numbers differ - expected: '3', found: '0'

Subtask #3:

score: 19
Accepted

Test #17:

score: 19
Accepted
time: 37ms
memory: 343948kb

input:

60 2 14 266401688520
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
712303980 712303980 712303979 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980 712303980...

output:

120

result:

ok 1 number(s): "120"

Test #18:

score: 19
Accepted
time: 28ms
memory: 344200kb

input:

56 2 7 534719494983
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
649719921 649719921 649719920 649719921 649719920 649719921 649719921 649719921 649719921 649719921 649719921 649719921 649719921 649719921 649719921 649719921 64971992...

output:

12620256

result:

ok 1 number(s): "12620256"

Test #19:

score: 19
Accepted
time: 32ms
memory: 343948kb

input:

59 2 18 362091866924
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
542053693 542053693 542053693 542053693 542053693 542053693 542053693 542053693 542053693 542053693 542053693 542053692 542053693 542053693 542053693 542053693 5...

output:

1037158320

result:

ok 1 number(s): "1037158320"

Subtask #4:

score: 0
Skipped

Dependency #1:

0%

Subtask #5:

score: 0
Skipped

Dependency #2:

0%

Subtask #6:

score: 0
Skipped

Dependency #1:

0%