QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#749370 | #2562. Fake Plastic Trees 2 | arvindr9 | WA | 1ms | 3812kb | C++20 | 4.4kb | 2024-11-14 23:54:10 | 2024-11-14 23:54:10 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#define DEBUG(...) debug(#__VA_ARGS__, __VA_ARGS__)
#else
#define DEBUG(...) 6
#endif
template<typename T, typename S> ostream& operator << (ostream &os, const pair<T, S> &p) {return os << "(" << p.first << ", " << p.second << ")";}
template<typename C, typename T = decay<decltype(*begin(declval<C>()))>, typename enable_if<!is_same<C, string>::value>::type* = nullptr>
ostream& operator << (ostream &os, const C &c) {bool f = true; os << "["; for (const auto &x : c) {if (!f) os << ", "; f = false; os << x;} return os << "]";}
template<typename T> void debug(string s, T x) {cerr << "\033[1;35m" << s << "\033[0;32m = \033[33m" << x << "\033[0m\n";}
template<typename T, typename... Args> void debug(string s, T x, Args... args) {for (int i=0, b=0; i<(int)s.size(); i++) if (s[i] == '(' || s[i] == '{') b++; else
if (s[i] == ')' || s[i] == '}') b--; else if (s[i] == ',' && b == 0) {cerr << "\033[1;35m" << s.substr(0, i) << "\033[0;32m = \033[33m" << x << "\033[31m | "; debug(s.substr(s.find_first_not_of(' ', i + 1)), args...); break;}}
typedef int int2;
#define __int128 long long
#define pi pair<int, int>
#define vi vector<int>
#define vii vector<vector<int>>
#define vpi vector<pi>
#define lep(i,l,r) for(int i=l;i<=r;++i)
#define rep(i,r,l) for(int i=r;i>=l;--i)
#define pb push_back
#define mp make_pair
#define eb emplace_back
#define f first
#define s second
int t;
const int maxn = 1005;
const int maxk = 51;
set<pi> dp[maxn]; // (num components, total overflow)
int2 main() {
ios::sync_with_stdio(0);
cin.tie(0);
cin >> t;
while (t--) {
int n, k, l, r;
{
long long a, b, c, d;
cin >> a >> b >> c >> d;
n = a;
k = b;
l = c;
r = d;
}
vector<int> a(n+1);
lep(i,1,n) {
long long x;
cin >> x;
a[i] = x;
}
vector<vector<int>> adj(n+1);
lep(i,1,n-1) {
long long u,v;
cin >> u >> v;
adj[u].pb(v);
adj[v].pb(u);
}
vector<int> sub(n+1);
auto dfs1 = [&](auto &self, int u, int p) -> void {
int val = a[u];
for (int v: adj[u]) {
if (v != p) {
self(self,v,u);
val += sub[v];
}
}
sub[u] = val;
// assert(sub[u] >= 0);
};
dfs1(dfs1,1,0);
auto dfs2 = [&](auto &self, int u, int p) -> void {
set<pi> curdp = {mp(0LL,0LL)};
for (int v: adj[u]) {
if (v != p) {
self(self,v,u);
set<pi> nextdp;
for (auto [nc, oc]: curdp) {
for (auto [nv, ov]: dp[v]) {
if (nc + nv <= k) {
nextdp.insert({nc+nv,oc+ov});
}
}
}
swap(curdp, nextdp);
}
}
// DEBUG(u,curdp);
for (auto [nc, oc]: curdp) {
// remaining stuff is <= R -> might be relevant for future components
int gap = sub[u] - l * nc - oc;
if (gap <= r) {
dp[u].insert({nc,oc});
}
// check if you're allowed to make a new component
if (l <= gap and gap <= r) {
if (nc == k+1) continue;
dp[u].insert({nc+1, sub[u] - (nc+1)*l});
}
}
};
dfs2(dfs2,1,0);
// DEBUG(dp[1]);
// DEBUG(dp[2]);
// DEBUG(dp[3]);
// DEBUG(dp[4]);
lep(i,1,k+1) {
int target_overflow = sub[1] - i * l;
// DEBUG(i,target_overflow);
if (dp[1].count({i,target_overflow})) {
cout << 1;
} else {
cout << 0;
}
}
cout << "\n";
lep(i,1,n) {
dp[i].clear();
}
}
}
/* stuff you should look for
* int overflow, array bounds
* special cases (n=1?)
* do smth instead of nothing and stay organized
* WRITE STUFF DOWN
* DON'T GET STUCK ON ONE APPROACH
*/
詳細信息
Test #1:
score: 100
Accepted
time: 0ms
memory: 3552kb
input:
3 4 3 1 2 1 1 1 1 1 2 2 3 3 4 4 3 1 2 1 1 1 1 1 2 1 3 1 4 4 3 0 0 1 1 1 1 1 2 1 3 1 4
output:
0111 0011 0000
result:
ok 3 tokens
Test #2:
score: 0
Accepted
time: 1ms
memory: 3600kb
input:
100 10 9 18 50 18 0 9 8 11 11 13 16 9 2 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 38 50 4 10 11 13 19 6 14 14 20 14 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 26 50 6 1 12 7 1 12 20 2 0 12 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 29 50 16 13 1 17 20 15 0 3 6 7 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10...
output:
0011000000 0010000000 0100000000 0010000000 0011111000 0111100000 0010000000 0010000000 0100000000 0011111000 0110000000 0011000000 0011111100 0100000000 0010000000 0010000000 0010000000 0011000000 0111000000 0011100000 0100000000 0100000000 0100000000 0011000000 0011000000 0011111000 0011111110 001...
result:
ok 100 tokens
Test #3:
score: 0
Accepted
time: 1ms
memory: 3656kb
input:
100 10 9 23 50 13 10 9 11 14 13 11 9 14 14 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 11 50 11 9 9 9 14 7 9 12 14 5 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 27 50 14 13 9 13 9 13 9 14 14 5 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 41 50 8 10 10 13 8 6 12 7 10 5 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 1...
output:
0011000000 0011110000 0010000000 0100000000 0011110000 0100000000 0011110000 0011110000 0011100000 0100000000 0011111100 0100000000 0011100000 0011100000 0100000000 0100000000 0011111000 0011000000 0100000000 0011000000 0011111100 0011100000 0100000000 0100000000 0100000000 0011111000 0011111111 010...
result:
ok 100 tokens
Test #4:
score: 0
Accepted
time: 1ms
memory: 3540kb
input:
100 10 9 17 50 9 8 10 12 12 10 12 10 9 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 19 50 10 9 8 10 12 12 8 11 12 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 46 50 8 8 10 10 10 8 9 10 11 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 9 50 9 8 11 10 11 10 10 11 11 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 ...
output:
0011100000 0011000000 0100000000 0011111110 0011111000 0011111111 0011111000 0100000000 0011110000 0011100000 0100000000 0011100000 0011100000 0100000000 0011110000 0011110000 0011100000 0100000000 0011100000 0011111111 0100000000 0011100000 0011000000 0100000000 0011111100 0011110000 0100000000 001...
result:
ok 100 tokens
Test #5:
score: 0
Accepted
time: 1ms
memory: 3656kb
input:
100 10 9 10 50 10 11 11 10 9 9 11 9 11 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 47 50 9 9 9 11 9 10 10 10 10 9 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 49 50 9 10 9 11 11 9 11 10 10 9 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 19 50 10 9 11 11 10 11 11 9 9 11 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10...
output:
0011111100 0100000000 0100000000 0011100000 0011110000 0011111111 0011110000 0100000000 0010000000 0100000000 0011111000 0011100000 0100000000 0011110000 0011110000 0011111000 0011111111 0011110000 0011111000 0011110000 0011111100 0011100000 0011000000 0011111111 0100000000 0011111111 0100000000 001...
result:
ok 100 tokens
Test #6:
score: 0
Accepted
time: 0ms
memory: 3556kb
input:
100 10 9 50 50 50 50 50 50 50 50 50 50 50 50 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 50 50 50 50 50 50 50 50 50 50 50 50 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 50 50 50 50 50 50 50 50 50 50 50 50 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 50 50 50 50 50 50 50 50 50 50 50 50 1 2 2 3 3 4 4 5 5 6 6...
output:
0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 0000000001 000...
result:
ok 100 tokens
Test #7:
score: -100
Wrong Answer
time: 1ms
memory: 3812kb
input:
100 10 9 1793281831312430 2579712950976669 883262428445148 926941407766037 874610589009581 918671242302849 917202064660727 848094660280817 810513162735522 862049976415823 844577745576407 873085228984439 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 10 9 1890762965379103 2701769025604615 804306683243252 81402...
output:
0000000000 0000000000 0000000000 0000000000 0000000000 0010000000 0000000000 0111100000 0000000000 1111100000 0000000000 0000000000 0000000000 0000000000 0011000000 0000000000 0000000000 0000000000 0000000000 0000000000 0000000000 0000000000 0000000000 0000000000 0111100000 0000000000 0000000000 000...
result:
wrong answer 1st words differ - expected: '0001000000', found: '0000000000'