Submission #1872992
Source Code Expand
#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)(n);++i)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;--i)
#define srep(i,s,t) for(int i=(int)(s);i<(int)(t);++i)
#define each(a,b) for(auto (a): (b))
#define all(v) (v).begin(),(v).end()
#define len(v) (int)(v).size()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define cmx(x,y) x=max(x,y)
#define cmn(x,y) x=min(x,y)
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define svec(v) cout<<#v<<":";rep(kbrni,v.size())cout<<" "<<v[kbrni];cout<<endl
#define sset(s) cout<<#s<<":";each(kbrni,s)cout<<" "<<kbrni;cout<<endl
#define smap(m) cout<<#m<<":";each(kbrni,m)cout<<" {"<<kbrni.first<<":"<<kbrni.second<<"}";cout<<endl
using namespace std;
typedef pair<int,int> P;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<double> vd;
typedef vector<P> vp;
typedef vector<string> vs;
const int MAX_N = 200005;
struct edge
{
int from,to;
ll cost;
bool operator< (const edge& another) const {
return cost > another.cost;
}
};
struct eda
{
int to;
ll cost;
};
vector<eda> G[MAX_N];
int id;
int nw[MAX_N];
class UF {
private:
int sz; vector<int> par,nrank,size;
public:
UF(){}
UF(int node_size){ sz = node_size; par.resize(sz),nrank.resize(sz),size.resize(sz);
rep(i,sz){ par[i] = i; nrank[i] = 0; size[i] = 1;} }
int find(int x){ if(par[x] == x){ return x; }else{ return par[x] = find(par[x]); } }
void unite(int x,int y,ll cost){
x = find(x),y = find(y);
if(x == y) return;
if(nrank[x] < nrank[y]) swap(x,y);
++id;
G[id].pb((eda){nw[x],cost*size[y]}),G[id].pb((eda){nw[y],cost*size[x]});
nw[x] = id;
par[y] = x; size[x] += size[y];
if(nrank[x] == nrank[y]) nrank[x]++;
}
int query(int x){ x = find(x); return size[x]; }
bool same(int x,int y){ return find(x) == find(y); }
};
ll sm[MAX_N];
void dfs(int u,ll dist){
sm[u] = dist;
rep(i,len(G[u])){
dfs(G[u][i].to,dist+G[u][i].cost);
}
}
int main()
{
cin.tie(0);
ios::sync_with_stdio(false);
int n;
cin >> n;
vector<edge> vec(n-1);
rep(i,n-1){
int a,b,c;
cin >> a >> b >> c;
vec[i] = (edge){a-1,b-1,c};
}
sort(all(vec));
rep(i,n){
nw[i] = i;
}
id = n-1;
UF uf(n);
rep(i,n-1){
uf.unite(vec[i].from,vec[i].to,vec[i].cost);
}
dfs(id,0);
rep(i,n){
cout << sm[i] << "\n";
}
return 0;
}
Submission Info
Submission Time |
|
Task |
E - Black Cats Deployment |
User |
kopricky |
Language |
C++14 (GCC 5.4.1) |
Score |
800 |
Code Size |
2872 Byte |
Status |
AC |
Exec Time |
71 ms |
Memory |
20736 KB |
Judge Result
Set Name |
Sample |
Subtask1 |
Subtask2 |
All |
Score / Max Score |
0 / 0 |
200 / 200 |
200 / 200 |
400 / 400 |
Status |
|
|
|
|
Set Name |
Test Cases |
Sample |
00_example_01.txt, 00_example_02.txt, 00_example_03.txt |
Subtask1 |
00_example_01.txt, 00_example_02.txt, 00_example_03.txt, s1_01.txt, s1_02.txt, s1_03.txt, s1_04.txt, s1_05.txt, s1_06.txt, s1_07.txt |
Subtask2 |
00_example_02.txt, s1_07.txt, s2_08.txt, s2_09.txt, s2_10.txt, s2_11.txt, s2_12.txt, s2_13.txt, s2_14.txt |
All |
00_example_01.txt, 00_example_02.txt, 00_example_03.txt, s1_01.txt, s1_02.txt, s1_03.txt, s1_04.txt, s1_05.txt, s1_06.txt, s1_07.txt, s2_08.txt, s2_09.txt, s2_10.txt, s2_11.txt, s2_12.txt, s2_13.txt, s2_14.txt, s3_15.txt, s3_16.txt, s3_17.txt, s3_18.txt, s3_19.txt, s3_20.txt, s3_21.txt, s3_22.txt, s3_23.txt, s3_24.txt, s3_25.txt, s3_26.txt, s3_27.txt, s3_28.txt, s3_29.txt |
Case Name |
Status |
Exec Time |
Memory |
00_example_01.txt |
AC |
3 ms |
5248 KB |
00_example_02.txt |
AC |
3 ms |
5248 KB |
00_example_03.txt |
AC |
2 ms |
5248 KB |
s1_01.txt |
AC |
3 ms |
5376 KB |
s1_02.txt |
AC |
3 ms |
5376 KB |
s1_03.txt |
AC |
3 ms |
5376 KB |
s1_04.txt |
AC |
3 ms |
5376 KB |
s1_05.txt |
AC |
3 ms |
5376 KB |
s1_06.txt |
AC |
3 ms |
5376 KB |
s1_07.txt |
AC |
2 ms |
5248 KB |
s2_08.txt |
AC |
51 ms |
14848 KB |
s2_09.txt |
AC |
26 ms |
9984 KB |
s2_10.txt |
AC |
16 ms |
8064 KB |
s2_11.txt |
AC |
58 ms |
16384 KB |
s2_12.txt |
AC |
58 ms |
16000 KB |
s2_13.txt |
AC |
58 ms |
15360 KB |
s2_14.txt |
AC |
53 ms |
15360 KB |
s3_15.txt |
AC |
61 ms |
15360 KB |
s3_16.txt |
AC |
31 ms |
10112 KB |
s3_17.txt |
AC |
19 ms |
8192 KB |
s3_18.txt |
AC |
31 ms |
10496 KB |
s3_19.txt |
AC |
71 ms |
16768 KB |
s3_20.txt |
AC |
70 ms |
16128 KB |
s3_21.txt |
AC |
69 ms |
16896 KB |
s3_22.txt |
AC |
69 ms |
16896 KB |
s3_23.txt |
AC |
70 ms |
15872 KB |
s3_24.txt |
AC |
64 ms |
15872 KB |
s3_25.txt |
AC |
61 ms |
20736 KB |
s3_26.txt |
AC |
63 ms |
15872 KB |
s3_27.txt |
AC |
62 ms |
20736 KB |
s3_28.txt |
AC |
63 ms |
15872 KB |
s3_29.txt |
AC |
61 ms |
20736 KB |