Submission #1077614

#	Time	Username	Problem	Language	Result	Execution time	Memory
1077614		vjudge1	Reconstruction Project (JOI22_reconstruction)	C++17	7 / 100	5098 ms	4944 KiB

reconstruction

#include <bits/stdc++.h>
#define F first
#define S second
#define pb push_back
#define ppb pop_back
#define ep insert
#define endl '\n'
#define elif else if
#define pow pwr
#define sqrt sqrtt
#define int long long
#define ll long long
typedef unsigned long long ull;
using namespace std;
struct edge{
	int a,b,w;
};
const int N=505,M=1e5+5;
int n,sz[N],par[N],m,q,W;
edge e[M];
bool cmp(edge x,edge y){
	return abs(x.w-W)<abs(y.w-W);
}
int getpar(int x){
	if (x==par[x]) return x;
	return par[x]=getpar(par[x]);
}
int link(int x,int y){
	x=getpar(x);
	y=getpar(y);
	if (x==y) return 0;
	if (sz[x]>sz[y]) swap(x,y);
	sz[y]+=sz[x];
	par[x]=y;
	return 1;
}
int32_t main(){
	ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
	cin>>n>>m;
	for (int i=1;i<=m;i++) cin>>e[i].a>>e[i].b>>e[i].w;
	cin>>q;
	while (q--){
		cin>>W;
		sort(e+1,e+1+m,cmp);
		for (int i=1;i<=n;i++) par[i]=i,sz[i]=1;
		int ans=0;
		for (int i=1;i<=m;i++) ans+=link(e[i].a,e[i].b)*abs(e[i].w-W);
		cout<<ans<<endl;
	}
	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