QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#761728 | #8237. Sugar Sweet II | Mine_King | TL | 3ms | 16016kb | C++14 | 1.7kb | 2024-11-19 09:38:28 | 2024-11-19 09:38:28 |
Judging History
answer
// Think twice, code once.
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
#define eputchar(c) putc(c, stderr)
#define eprintf(...) fprintf(stderr, __VA_ARGS__)
#define eputs(str) fputs(str, stderr), putc('\n', stderr)
using namespace std;
const int mod = 1e9 + 7;
int T, n, a[500005], p[500005], w[500005], q[500005];
int fac[500005], inv[500005];
int ans[500005];
void solve(int st) {
ans[st] = 1;
int now = q[st];
while (now != st) {
if (a[now] < a[p[now]] || a[now] >= a[p[now]] + w[p[now]]) break;
ans[now] = ans[p[now]] + 1;
now = q[now];
}
for (int j = q[st]; j != now; j = q[j]) ans[j] = inv[ans[j]] % mod;
return;
}
int main() {
fac[0] = fac[1] = 1;
inv[0] = inv[1] = 1;
for (int i = 2; i <= 500000; i++) {
fac[i] = (long long)fac[i - 1] * i % mod;
inv[i] = (long long)(mod - mod / i) * inv[mod % i] % mod;
}
for (int i = 2; i <= 500000; i++) inv[i] = (long long)inv[i - 1] * inv[i] % mod;
scanf("%d", &T);
while (T--) {
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
for (int i = 1; i <= n; i++) scanf("%d", &p[i]), q[p[i]] = i;
for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
for (int i = 1; i <= n; i++) ans[i] = -1;
for (int i = 1; i <= n; i++)
if (ans[i] == -1 && a[i] >= a[p[i]] + w[p[i]]) {
ans[i] = 0;
if (p[i] == i) continue;
for (int j = q[i]; j != i; j = q[j])
if (ans[j] != -1 && a[j] >= a[p[j]]) ans[j] = 0;
else break;
} else if (ans[i] == -1 && a[i] < a[p[i]]) solve(i);
for (int i = 1; i <= n; i++)
if (ans[i] == -1) ans[i] = 0;
for (int i = 1; i <= n; i++) printf("%lld ", (a[i] + (long long)ans[i] * w[i]) % mod);
puts("");
}
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 3ms
memory: 16016kb
input:
4 4 2 5 5 2 4 2 1 3 3 2 1 4 3 5 4 3 1 1 1 6 6 6 3 5 4 3 2 3 1 1 2 3 5 2 1 3 2 1 5 1 1 3 4 1 3 4 2 4
output:
500000007 5 5 6 5 10 9 166666673 5 6 500000006 4 3 4 5
result:
ok 15 numbers
Test #2:
score: -100
Time Limit Exceeded
input:
50000 5 508432375 168140163 892620793 578579275 251380640 3 4 4 1 3 346232959 736203130 186940774 655629320 607743104 1 863886789 1 364158084 18 864679185 463975750 558804051 604216585 694033700 499417132 375390750 337590759 467353355 111206671 983760005 984444619 322277587 138763925 205122047 97736...