Untitled

 avatar
unknown
plain_text
a year ago
3.0 kB
8
Indexable
#include<bits/stdc++.h>

#define ll long long
#define pp push_back
//#define endl '\n'
#define all(x) x.begin(),x.end()
#define ld long double
#define PI acos(-1)
#define sin(a) sin((a)*PI/180)
#define cos(a) cos((a)*PI/180)
#define ones(x) __builtin_popcountll(x)
//#define int ll

using namespace std;

void Drakon() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
#ifdef Clion
    freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
#endif
}

unsigned long long inf = 1e10;
const double EPS = 1e-6;
const int MOD = 1000000007, N = 200005, LOG = 25;

ll mul(const ll &a, const ll &b) {
    return (a % MOD + MOD) * (b % MOD + MOD) % MOD;
}

ll add(const ll &a, const ll &b) {
    return (a + b + 2 * MOD) % MOD;
}

ll pw(ll x, ll y) {
    ll ret = 1;
    while (y > 0) {
        if (y % 2 == 0) {
            x = mul(x, x);
            y = y / 2;
        } else {
            ret = mul(ret, x);
            y = y - 1;
        }
    }
    return ret;
}

ll ask(int a, int b, int c, int d){
    cout << "? " << a + 1 << " " << b + 1 << " " << c + 1 << " " << d + 1 << endl;
    ll ans;
    cin >> ans;
    return ans;
}

int n;
ll k;
vector<pair<int, ll>> adj[N];
map<pair<int, int>, ll> cost;

void dfs(int u, int p1, int p2, ll edge) {

    ll ret = ask(p2, u, p2, p1) - 2 * edge;
    cost[{u, p1}] = cost[{p1, u}] = ret;

    for(auto v : adj[u]){
        if(v.first == p1)continue;
        dfs(v.first, u, p1, ret);
    }
}

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

    int root = 0;
    for (int i = 0; i < n; ++i) {
        if(adj[i].size() > 1){
            root = i;
            break;
        }
    }

    ll r1 = ask(adj[root][0].first, adj[root][1].first, adj[root][0].first, root);
    ll r2 = ask(adj[root][0].first, adj[root][1].first, root, adj[root][1].first);
    ll sum = (r1 + r2) / 3;
    cost[{root, adj[root][0].first}] = cost[{adj[root][0].first, root}] = r1 - sum;
    cost[{root, adj[root][1].first}] = cost[{adj[root][1].first, root}] = r2 - sum;
    for (int i = 2; i < adj[root].size(); ++i) {
        dfs(adj[root][i].first, root, adj[root][1].first, r2 - sum);
    }

    for(auto v : adj[adj[root][0].first]){
        if(v.first == root)continue;
        dfs(v.first, adj[root][0].first, root, r1 - sum);
    }
    for(auto v : adj[adj[root][1].first]){
        if(v.first == root)continue;
        dfs(v.first, adj[root][1].first, root, r2 - sum);
    }

    for (int i = 0; i < n; ++i) {
        for(auto &v : adj[i]){
            v.second = cost[{i, v.first}];
        }
    }


}

signed main() {
    Drakon();
    int t = 1;
    cin >> t;
    while (t--) {
        solve();
    }
}
Editor is loading...
Leave a Comment