QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#711239 | #4781. 完美的集合 | NineSuns | 0 | 92ms | 344224kb | C++14 | 5.7kb | 2024-11-05 07:31:02 | 2024-11-05 07:31:02 |
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]) {
if (i.fi == fa) continue;
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};
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) {
gs[j+z] = gs[j+z]+(fs[p][j]*fs[i.fi][z]);
}
}
memcpy(fs[p], gs, sizeof gs);
}
}
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 lC[25][25];
struct vc {
LL f[25];
void clr () { memset(f, 0, sizeof f); }
friend vc operator + (const vc & a, const ll &b) {
vc c; c.clr();
for (int i = 0;i < 25;i++) {
LL cb = 1;
for (int j = 0;j <= i;j++) c.f[i-j] += a.f[i]*cb%mod*lC[i][j]%mod, c.f[i-j] %= mod, cb = cb*b%mod;
}
return c;
}
friend vc operator * (const vc & a, const vc & b) {
vc c; memset(c.f, 0, sizeof c.f);
for (int i = 0;i < 25;i++) {
for (int j = 0;i+j < 25;j++) c.f[i+j] += a.f[i]*b.f[j]%mod, c.f[i+j] %= mod;
}
return c;
}
}dd;
void init () {
dd.clr();
for (int i = 0;i < 25;i++) {
lC[i][0] = 1;
for (int j = 1;j <= i;j++) lC[i][j] = (lC[i-1][j]+lC[i][j-1])%mod;
}
for (int i = 0;i < 16;i++) {
ll k = 1;
for (int j = 0;j < 4;j++) if (i>>j&1) k = k*(j+1);
dd.f[4-__builtin_popcount(i)] += k;
}
}
vc gv (ll n) {
if (n == 0) return dd;
vc k = gv(n-1>>1);
vc res = k*(k+(((n-1)/2+1)*5%mod));
if (n&1^1) res = res*(dd+(n*5%mod));
return res;
}
LL sc (ll n) {
if (n < 0) return 1;
return gv(n).f[0];
}
LL fac (ll n) {
if (n == 0) return 1;
LL res = 1;
for (ll i = n/5*5+1;i <= n;i++) res = res*i%mod;
res = res*fac(n/5)%mod*sc(n/5-1)%mod;
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;
return fac(n)*qpow(fac(n-m)*fac(m)%mod)%mod*x%mod;
}
void solve () {
init();
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++) if (f[i][0][j] > ans) ans = f[i][0][j]; //, cout << "DP:" << i << " " << j << " " << f[i][0][j] << endl;
// cout << ans << endl;
for (int i = 1;i <= n;i++) {
memset(sk, 0, sizeof sk);
for (int j = 1;j <= n;j++) vis[j] = chk(i, j); //, cout << vis[j] << " "; cout << endl;
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 () {
// freopen("t.out", "w", stdout);
// 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: 7ms
memory: 339852kb
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: 15ms
memory: 339840kb
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: 0
Wrong Answer
time: 16ms
memory: 338300kb
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:
2616789430297637
result:
wrong answer 1st numbers differ - expected: '12271512', found: '2616789430297637'
Subtask #2:
score: 0
Wrong Answer
Test #13:
score: 11
Accepted
time: 20ms
memory: 341900kb
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:
3
result:
ok 1 number(s): "3"
Test #14:
score: 11
Accepted
time: 23ms
memory: 342024kb
input:
39 617 1 172 60 66 69 46 26 120 82 68 86 29 11 115 56 36 97 89 43 101 92 25 57 45 26 1 111 67 93 84 4 74 36 67 63 60 64 83 104 5 110 2 2 2 2 2 2 2 2 2 1 2 2 2 2 2 1 2 2 1 2 2 2 2 2 2 2 2 2 2 1 2 2 1 2 2 2 2 2 2 1 2 52 1 3 22 1 4 50 3 5 19 1 6 26 1 7 48 1 8 30 3 9 55 2 10 18 5 11 28 5 12 19 1 13 55 1...
output:
3
result:
ok 1 number(s): "3"
Test #15:
score: 0
Wrong Answer
time: 92ms
memory: 344224kb
input:
59 9134 1 30156516 1838 15 2444 1837 3666 1840 2447 1835 10 3057 9 2444 13 3662 13 1836 3057 2448 3055 12 15 11 8 4271 9 12 11 3055 1835 3057 2446 2449 2446 12 3663 12 2440 8 3666 1837 1837 3057 9 2447 3057 1835 1837 2446 1838 2442 2443 2447 2449 1835 3058 11 4274 3664 2441 13038 0 17384 13037 7589 ...
output:
3017131417650129
result:
wrong answer 1st numbers differ - expected: '129', found: '3017131417650129'
Subtask #3:
score: 0
Wrong Answer
Test #17:
score: 0
Wrong Answer
time: 28ms
memory: 343952kb
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:
11015843440008245
result:
wrong answer 1st numbers differ - expected: '120', found: '11015843440008245'
Subtask #4:
score: 0
Skipped
Dependency #1:
0%
Subtask #5:
score: 0
Skipped
Dependency #2:
0%
Subtask #6:
score: 0
Skipped
Dependency #1:
0%