QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#626508#9110. Zayin and Treehuangzihao123WA 508ms17464kbC++202.8kb2024-10-10 09:49:112024-10-10 09:49:11

Judging History

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

  • [2024-10-10 09:49:11]
  • 评测
  • 测评结果:WA
  • 用时:508ms
  • 内存:17464kb
  • [2024-10-10 09:49:11]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define io ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define all(a) a.begin(),a.end()
typedef pair<int,int> PII;
typedef long long LL;
const int N = 2e5 + 10;
const int mod = 998244353;
inline int ls(int x) { return x << 1; }
inline int rs(int x) { return x << 1 | 1; }
int dx[8] = { 1,0,0,-1,-1,-1,1,1 };
int dy[8] = { 0,-1,1,0,-1,1,-1,1};
inline int gcd(int a, int b)
{
   if (b) while ((a %= b) && (b %= a));
   return a + b;
}
void solve()
{
    int n;
    cin >> n;
    vector<int>a(n + 1);
    vector<vector<int>>g(n + 1);
    for(int i = 1;i <= n;i++) cin >> a[i];
    for(int i = 1;i < n;i++){
        int u,v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    int minpart = 0,root = 0;
    vector<int>sz(n + 1),buc(n + 1),dep(n + 1),mi(n + 1),mx(n + 1);
    for(int i = 1;i <= n;i++){
        mi[i] = mx[i] = a[i];
    }
    vector<bool>vis(n + 1);
    int tol = 0;
    auto getroot = [&](auto getroot,int sum,int u,int fa) -> void { 
        sz[u] = 1;
        int maxpart = 0;
        for(int v : g[u]){
            if(v == fa || vis[v]) continue;
            getroot(getroot,sum,v,u);
            sz[u] += sz[v];
            maxpart = max(maxpart,sz[v]);
        }
        maxpart = max(maxpart,sum - sz[u]);
        if(maxpart < minpart){
            minpart = maxpart;
            root = u;
        }
    };
    auto dfs = [&](auto dfs,int u,int fa) -> void {
        buc[++tol] = u;
        dep[u] = dep[fa] + 1;
        mx[u] = max(mx[fa],mx[u]);
        mi[u] = min(mi[fa],mi[u]);
        for(int v : g[u]){
            if(v == fa || vis[v]) continue;
            dfs(dfs,v,u);
        }
    };
    int ans = 1e18;
    auto sol = [&](auto sol,int sum,int u,int fa) -> void {
        minpart = sum;
        getroot(getroot,sum,u,fa);
        vis[root] = 1;
        dep[root] = 1;
        mi[root] = a[root];
        int res = 1e18;
        for(int v : g[root]){
            if(v == fa || vis[v]) continue;
            tol = 0;
            dfs(dfs,v,root);
            for(int i = 1;i <= tol;i++){
                int u = buc[i];
                ans = min({ans,mi[u] + dep[u] - max(mx[u],a[root]),mi[u] + dep[u] + res});
            }
            for(int i = 1;i <= tol;i++){
                int u = buc[i];
                res = min(res,dep[u] - 1 + mx[u]);
            }
        }
        for(auto v : g[root]){
            if(v == fa || vis[v]) continue;
            sol(sol,sz[v],v,root);
        }
    };
    sol(sol,n,1,0);
    cout << ans << endl;
}
signed main()
{
    io;
    int tt;
    cin >> tt;
    while(tt--){
        solve();
    }
	return 0;
}
/*

*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Wrong Answer
time: 508ms
memory: 17464kb

input:

3009
5
4 5 3 4 2
1 2
2 3
3 4
3 5
5
4 4 1 1 2
1 2
2 3
3 4
3 5
10
5 8 1 0 8 7 5 2 0 4
2 4
3 8
3 9
1 2
1 3
3 6
4 5
5 7
6 10
10
6 8 8 4 8 0 6 6 0 2
7 10
1 7
2 9
2 3
3 4
1 5
1 6
6 8
1 2
10
9 0 4 0 4 6 0 2 0 0
1 5
1 3
1 7
2 6
1 2
1 9
1 4
5 8
7 10
10
8 8 1 2 7 4 8 6 0 8
1 6
1 7
1 5
7 9
1 3
1 2
2 10
3 4
1 8...

output:

0
-1
-6
-6
-7
-6
-7
-4
-3
-7
-5
-6
-2
-4
-6
-4
-4
-7
-4
-4
-6
-6
-3
-4
-5
-6
-6
-6
-7
-7
-6
-7
-6
-6
-7
-7
-5
-6
-4
-7
-4
-4
-6
-7
-6
-6
-3
-6
-4
-6
-4
-6
-7
-6
-7
-4
-6
-5
-7
-6
-4
-7
-3
-3
-5
-6
-2
-4
-7
-6
-5
-6
-4
-3
-6
-2
-4
-2
-6
-5
-7
-4
-5
-3
-7
-7
-4
-6
-5
-4
-6
-5
-4
-6
-4
-7
-7
-7
-7
-5
-...

result:

wrong answer 13th lines differ - expected: '-5', found: '-2'