#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

int n;
vector<long long> a;
vector<vector<int>> adj;
vector<int> subtree_size;
long long total_good_triplets;

// 检查是否为完美平方数
bool is_perfect_square(long long x) {
    long long r = round(sqrt(x));
    return r * r == x;
}

void dfs(int u, int p) {
    subtree_size[u] = 1;
    vector<long long> comp_sizes;
    
    for (int v : adj[u]) {
        if (v == p) continue;
        dfs(v, u);
        subtree_size[u] += subtree_size[v];
        comp_sizes.push_back(subtree_size[v]); // 子树构成一个连通块
    }
    
    if (u != 1) {
        comp_sizes.push_back(n - subtree_size[u]); // 父亲方向的剩余部分构成一个连通块
    }
    
    // 如果当前节点是完美平方数，统计以它为核心的中位数三元组
    if (is_perfect_square(a[u])) {
        long long S1 = 0, S2 = 0, S3 = 0;
        for (long long s : comp_sizes) {
            S1 += s;
            S2 += s * s;
            S3 += s * s * s;
        }
        
        // E2 = sum_{a < b} s_a * s_b
        long long E2 = (S1 * S1 - S2) / 2;
        // E3 = sum_{a < b < c} s_a * s_b * s_c
        long long E3 = (S1 * S1 * S1 - 3 * S1 * S2 + 2 * S3) / 6;
        
        total_good_triplets += (E2 + E3);
    }
}

void solve() {
    cin >> n;
    a.resize(n + 1);
    adj.assign(n + 1, vector<int>());
    subtree_size.resize(n + 1);
    total_good_triplets = 0;
    
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    for (int i = 0; i < n - 1; ++i) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    
    dfs(1, 0);
    cout << total_good_triplets << "\n";
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int t;
    cin >> t;
    while (t--) {
        solve();
    }
    
    return 0;
}

#include <iostream>
#include <vector>
#include <numeric>
#include <algorithm>
#include <string>

using namespace std;

void print128(__int128 num) {
    if (num == 0) {
        cout << 0 << "\n";
        return;
    }
    string buf;
    while (num > 0) {
        buf += (char)('0' + (num % 10));
        num /= 10;
    }
    reverse(buf.begin(), buf.end());
    cout << buf << "\n";
}

void solve() {
    int arr_len;
    if (!(cin >> arr_len)) return;
    vector<long long> arr(arr_len + 1);
    for (int pos = 1; pos <= arr_len; ++pos) {
        cin >> arr[pos];
    }
    
    __int128 sum_total = 0;
    vector<pair<long long, int>> seg_gcd_list;
    
    for (int left_pt = arr_len; left_pt >= 1; --left_pt) {
        long long base_val = arr[left_pt];
        vector<pair<long long, int>> new_seg_gcd;
        long long curr_gcd = base_val;
        int last_seg_r = left_pt;
        
        for (auto& seg_item : seg_gcd_list) {
            long long new_g = gcd(curr_gcd, seg_item.first);
            if (new_g == curr_gcd) {
                last_seg_r = seg_item.second;
            } else {
                new_seg_gcd.push_back({curr_gcd, last_seg_r});
                curr_gcd = new_g;
                last_seg_r = seg_item.second;
            }
        }
        new_seg_gcd.push_back({curr_gcd, last_seg_r});
        seg_gcd_list.swap(new_seg_gcd);
        
        vector<pair<int, long long>> split_point_list;
        int seg_start = left_pt;
        for (int idx = 0; idx < (int)seg_gcd_list.size(); ++idx) {
            if (idx > 0) {
                split_point_list.push_back({seg_start, seg_gcd_list[idx - 1].first});
            }
            seg_start = seg_gcd_list[idx].second + 1;
        }
        
        vector<pair<long long, long long>> interval_set;
        int curr_r_begin = left_pt;
        
        for (int sp_idx = 0; sp_idx <= (int)split_point_list.size(); ++sp_idx) {
            int next_r_begin = (sp_idx < (int)split_point_list.size()) ? split_point_list[sp_idx].first : arr_len + 1;
            
            long long block_length = next_r_begin - curr_r_begin;
            if (block_length > 0) {
                long long min_diff = 0;
                if (!interval_set.empty()) {
                    vector<pair<long long, long long>> sorted_intervals = interval_set;
                    sort(sorted_intervals.begin(), sorted_intervals.end());
                    
                    long long res_min = 2e18;
                    long long max_right = base_val;
                    for (int i = 0; i < (int)sorted_intervals.size(); ++i) {
                        res_min = min(res_min, max_right - sorted_intervals[i].first);
                        max_right = max(max_right, sorted_intervals[i].second);
                    }
                    res_min = min(res_min, max_right - base_val);
                    min_diff = res_min;
                }
                sum_total += (__int128)min_diff * block_length;
            }
            
            if (sp_idx < (int)split_point_list.size()) {
                int split_pos = split_point_list[sp_idx].first;
                long long gcd_split = split_point_list[sp_idx].second;
                long long num_val = arr[split_pos];
                long long rem_diff = (base_val % gcd_split - num_val % gcd_split + gcd_split) % gcd_split;
                if (rem_diff > 0) {
                    long long l_bound = base_val - rem_diff;
                    long long r_bound = base_val - rem_diff + gcd_split;
                    interval_set.push_back({l_bound, r_bound});
                }
            }
            curr_r_begin = next_r_begin;
        }
    }
    print128(sum_total);
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int test_cnt;
    if (cin >> test_cnt) {
        while (test_cnt--) {
            solve();
        }
    }
    return 0;
}

// using namespace std;

// bool isSquare(long long x){
//     if(x < 0) return false;
//     long long r = round(sqrt(x));
//     return r * r == x;
// }

// void solve(){
//     int n;
//     cin >> n;

//     vector<long long> a(n + 1);
//     for(int i = 1; i <= n; i++){
//         cin >> a[i];
//     }

//     vector<vector<int>> adj(n + 1);
//     for(int i = 0; i < n - 1; i++){
//         int u, v;
//         cin >> u >> v;
//         adj[u].push_back(v);
//         adj[v].push_back(u);
//     }

//     vec
// }

// int main()
// {
//     ios_base::sync_with_stdio(false);
//     cin.tie(NULL);
//     int t;
//     cin >> t;
//     while (t--)
//     {

//     }
// }

#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

// Fast check for perfect square
bool is_square(long long x) {
    if (x < 0) return false;
    long long r = round(sqrt(x));
    return r * r == x;
}

void solve() {
    int n;
    cin >> n;
    
    vector<long long> a(n + 1);
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }

    vector<vector<int>> adj(n + 1);
    for (int i = 0; i < n - 1; ++i) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    vector<int> sz(n + 1, 0);
    long long total_good_triplets = 0;

    // DFS to calculate subtree sizes and count valid triplets
    auto dfs = [&](auto& self, int u, int p) -> void {
        sz[u] = 1;
        vector<int> component_sizes;
        
        for (int v : adj[u]) {
            if (v != p) {
                self(self, v, u);
                sz[u] += sz[v];
                component_sizes.push_back(sz[v]);
            }
        }
        
        // Add the component representing the rest of the tree (if u is not the root)
        if (n - sz[u] > 0) {
            component_sizes.push_back(n - sz[u]);
        }

        // If the current node is a valid median (its value is a perfect square)
        if (is_square(a[u])) {
            long long A = 0; // Ways to pick 3 nodes from different components
            long long B = 0; // Ways to pick 2 nodes from different components (3rd node is u)
            long long sum = 0;
            long long sum_pairs = 0;
            
            // Calculate combinations dynamically to maintain O(degree) complexity
            for (long long s : component_sizes) {
                A += sum_pairs * s;
                sum_pairs += sum * s;
                B += sum * s;
                sum += s;
            }
            total_good_triplets += (A + B);
        }
    };

    dfs(dfs, 1, 0);
    cout << total_good_triplets << "\n";
}

int main() {
    // Optimize standard I/O operations for competitive programming
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int t;
    cin >> t;
    while (t--) {
        solve();
    }
    
    return 0;
}

I completely understand how frustrating that is! My previous fix strictly resolved the C# compilation error (the variable naming conflict) so your code could finally compile and run, but it left your underlying mathematical logic untouched.

Now that the code is executing, the "Wrong Answer" is exposing a very subtle **sign error** in your Convex Hull Trick query for the left side (`tr2`).

### The Math Behind the Bug

For the right side (`tr1`), you want to maximize the score. Mathematically, the expression you are optimizing for index $$$i$$$ is:




<center>$$$s[i] - s[m] - (i - m) \cdot h[m]$$$</center>



Which expands and groups to:




<center>$$$(s[i]) - (h[m]) \cdot i - s[m] + m \cdot h[m]$$$</center>



Your `Query` function evaluates $$$y - M \cdot x$$$. For `tr1`, $$$y = s[i]$$$, $$$x = i$$$, and you correctly pass $$$M = h[m]$$$. This flawlessly matches the equation.

**However, for the left side (`tr2`)**, you want to maximize the score going backwards. That expression is:




<center>$$$s[m-1] - s[i-1] - (m - i) \cdot h[m]$$$</center>



Let's expand and group this one:




<center>$$$(-s[i-1]) + h[m] \cdot i + s[m-1] - m \cdot h[m]$$$</center>



Notice the **$$$+ h[m] \cdot i$$$**.
Since your `Query` function subtracts the slope term ($$$y - M \cdot x$$$), to get a positive $$$h[m] \cdot i$$$, you **must** pass $$$-h[m]$$$ as the slope. In your code, you passed a positive `h[m]`:

```csharp
// WRONG
long mxvl = tr2.Query(1, 1, n, a2, b2, h[m]) + s[m - 1] - (long)m * h[m];

```

This forces the tree to evaluate $$$-s[i-1] - h[m] \cdot i$$$, which completely breaks the calculation and underestimates the score.

### The Fix

Change `h[m]` to `-h[m]` in the `tr2.Query` call.

Here is the fully corrected code with the mathematical fix applied.

```csharp
// Author => Gopal

using System;
using System.IO;

#pragma warning disable CS8618 

class GRoadworks
{
    struct Point
    {
        public long x, y;
    }

    class ConvexHST
    {
        private Point[] hull;
        private int[] hullst;
        private int[] hullln;
        private int hulllptr;
        public ConvexHST(int n)
        {
            hull = new Point[n * 20];
            hullst = new int[4 * n + 5];
            hullln = new int[4 * n + 5];
        }
        public void clear()
        {
            hulllptr = 0;
        }
        private bool Bad(Point a, Point b, Point c)
        {
            Int128 dx1 = b.x - a.x;
            Int128 dy1 = b.y - a.y;
            Int128 dx2 = c.x - a.x;
            Int128 dy2 = c.y - a.y;
            return dx1 * dy2 - dy1 * dx2 >= 0;
        }

        public void build(int nd, int l, int r, long[] x, long[] y)
        {
            hullst[nd] = hulllptr;
            int len = 0;
            int strt = hulllptr;
            for (int i = l; i <= r; i++)
            {
                Point p = new Point { x = x[i], y = y[i] };
                while (len >= 2)
                {
                    Point a = hull[strt + len - 2];
                    Point b = hull[strt + len - 1];
                    if (Bad(a, b, p)) len--;
                    else break;
                }
                hull[strt + len] = p;
                len++;
            }
            hullln[nd] = len;
            hulllptr += len;
            if (l == r) return;
            int mid = l + (r-l) / 2;
            build(2 * nd, l, mid, x, y);
            build(2 * nd + 1, mid + 1, r, x, y);
        }

        public long Query(int nd, int l, int r, int ql, int qr, long m)
        {
            if (ql <= l && r <= qr)
            {
                int lt = 0, rt = hullln[nd] - 1;
                int strt = hullst[nd];
                while (lt < rt)
                {
                    int mid = lt + (rt - lt) / 2;
                    Point p1 = hull[strt + mid];
                    Point p2 = hull[strt + mid + 1];
                    if (p2.y - p1.y >= m * (p2.x - p1.x)) lt = mid + 1;
                    else  rt = mid;
                }
                return hull[strt + lt].y - m * hull[strt + lt].x;
            }
            int mdidx = l + (r - l) / 2;
            long res = long.MinValue;
            if (ql <= mdidx) res = Math.Max(res, Query(2 * nd, l, mdidx, ql, qr, m));
            if (qr > mdidx) res = Math.Max(res, Query(2 * nd + 1, mdidx + 1, r, ql, qr, m));
            return res;
        }
    }

    class FastIO
    {
        private Stream s;
        private byte[] buf = new byte[1 << 20];
        private int pos, len;
        public FastIO(Stream strm) { s = strm; }
        private int NextChar()
        {
            if (pos >= len)
            {
                pos = 0;
                len = s.Read(buf, 0, buf.Length);
                if (len == 0) return -1;
            }
            return buf[pos++];
        }

        public long NextLong()
        {
            int c = NextChar();
            while (c != -1 && c <= 32) c = NextChar();
            if (c == -1) return 0;
            long res = 0;
            while (c > 32)
            {
                res = res * 10 + (c - '0');
                c = NextChar();
            }
            return res;
        }
        public int NextInt() => (int)NextLong();
    }

    static void Main()
    {
        using var strm = new BufferedStream(Console.OpenStandardInput(), 1 << 20);
        FastIO io = new FastIO(strm);
        using var wrt = new StreamWriter(Console.OpenStandardOutput(), bufferSize: 1 << 20);
        int t = io.NextInt();
        if (t == 0) return;
        int mxn = 200005;
        long[] h = new long[mxn];
        long[] d = new long[mxn];
        long[] e = new long[mxn];
        long[] pe = new long[mxn];
        long[] s = new long[mxn];
        long[] xs = new long[mxn];
        long[] y1 = new long[mxn];
        long[] y2 = new long[mxn];
        ConvexHST tr1 = new ConvexHST(mxn);
        ConvexHST tr2 = new ConvexHST(mxn);
        while (t-- > 0)
        {
            int n = io.NextInt();
            long k = io.NextLong();
            int x = io.NextInt();
            for (int i = 1; i <= n; i++) h[i] = io.NextLong();
            for (int i = 1; i < n; i++) d[i] = io.NextLong();
            e[x] = 0;
            for (int i = x + 1; i <= n; i++)  e[i] = Math.Max(e[i - 1] + 1, d[i - 1]);
            for (int i = x - 1; i >= 1; i--) e[i] = Math.Max(e[i + 1] + 1, d[i]);
            int lmx = x, rmx = x;
            while (lmx > 1 && e[lmx - 1] <= k) lmx--;
            while (rmx < n && e[rmx + 1] <= k) rmx++;
            
            pe[x] = 0;
            long crmx = h[x];
            for (int i = x + 1; i <= n; i++)
            {
                if (e[i] > k) break;
                long w = e[i] - e[i-1] - 1;
                pe[i] = pe[i-1] + h[i] + w * crmx;
                crmx = Math.Max(crmx, h[i]);
            }
            crmx = h[x];
            for (int i = x - 1; i >= 1; i--)
            {
                if (e[i] > k) break;
                long w = e[i] - e[i + 1] - 1;
                pe[i] = pe[i + 1] + h[i] + w * crmx;
                crmx = Math.Max(crmx, h[i]);
            }
            s[0] = 0;
            for (int i = 1; i <= n; i++)
            {
                s[i] = s[i - 1] + h[i];
                xs[i] = i;
                y1[i] = s[i];
                y2[i] = -s[i - 1];
            }
            tr1.clear();
            tr1.build(1, 1, n, xs, y1);
            tr2.clear();
            tr2.build(1, 1, n, xs, y2);
            long ans = 0;
            for (int m = 1; m <= n; m++)
            {
                if (e[m] > k) continue;
                long bsscr = pe[m] + (k - e[m]) * h[m];
                int a1 = m;
                int b1 = (int)Math.Min(rmx, m + k - e[m]);
                if (a1 <= b1)
                {
                    long mxvl = tr1.Query(1, 1, n, a1, b1, h[m]) - s[m] + (long)m * h[m];
                    ans = Math.Max(ans, bsscr + mxvl);
                }
                int a2 = (int)Math.Max(lmx, m - (k - e[m]));
                int b2 = m;
                if (a2 <= b2)
                {
                    long mxvl = tr2.Query(1, 1, n, a2, b2, -h[m]) + s[m - 1] - (long)m * h[m];
                    ans = Math.Max(ans, bsscr + mxvl);
                }
            }
            wrt.WriteLine(ans);
        }
    }
}

```