QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#708794 | #4781. 完美的集合 | NineSuns | 19 | 37ms | 344200kb | C++14 | 4.9kb | 2024-11-04 06:44:14 | 2024-11-04 06:44:14 |
Judging History
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%