Submission #776892

#	Time	Username	Problem	Language	Result	Execution time	Memory
776892		eltu0815	Reconstruction Project (JOI22_reconstruction)	C++14	7 / 100	5069 ms	18964 KiB

reconstruction

#include <bits/stdc++.h>
#define MAX 500005
#define MOD (ll)(1e9+7)
#define INF (ll)(1e18)
 
using namespace std;    
typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
 
int n, m, q;
int parent[505];
vector<pair<int, pii> > edge;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    
    cin >> n >> m;
    for(int i = 1; i <= m; ++i) {
        int a, b, w; cin >> a >> b >> w;
        edge.push_back({w, {a, b}});
    }
    sort(edge.begin(), edge.end());
    
    auto getParent = [&](auto&& self, int node) -> int {
        if(parent[node] == node) return node;
        else return parent[node] = self(self, parent[node]);
    };
    
    auto unionParent = [&](int u, int v) -> void {
        u = getParent(getParent, u), v = getParent(getParent, v);
        if(u > v) swap(u, v);
        parent[v] = u;
    };
    
    cin >> q;
    while(q--) {
        int x; cin >> x;
        for(int i = 1; i <= n; ++i) parent[i] = i;
        ll ans = 0; pair<int, pii> pp = {x, {0, 0}};
        int q = lower_bound(edge.begin(), edge.end(), pp) - edge.begin(), p = q - 1;
        while(0 <= p || q < m) {
            ll i;
            if(p == -1) i = q++;
            else if(q == m) i = p--;
            else if(abs(edge[p].first - x) < abs(edge[q].first - x)) i = p--;
            else i = q++;
            
            if(getParent(getParent, edge[i].second.first) != getParent(getParent, edge[i].second.second)) {
                ans += 1ll * abs(edge[i].first - x);
                unionParent(edge[i].second.first, edge[i].second.second);
            }
        }
        cout << ans << '\n';
    }
    return 0;
}

• Subtask 1: 3 / 3
• Subtask 2: 4 / 4
• Subtask 3: 0 / 7
• Subtask 4: 0 / 28
• Subtask 5: 0 / 35
• Subtask 6: 0 / 23