Submission #400600

#	Time	Username	Problem	Language	Result	Execution time	Memory
400600		fadi57	Sjekira (COCI20_sjekira)	C++14	0 / 110	2 ms	1740 KiB

sjekira

#include<bits/stdc++.h>
using namespace std;
const int mxx=300+10;
typedef long long ll;
int inf=1e9+10;
const int mod=1e9+7;
int a[mxx];
ll ans=0;

pair<int,int>edge[mxx];
int n;int par[mxx];
int mx[mxx];
int siz[mxx];
vector<int>adj[mxx];
void init(){
for(int i=0;i<n;i++){

    par[i]=i;
    mx[i]=a[i];
    siz[i]=1;
}

}
int fin(int node){

if(par[node]!=node){

    return par[node]=fin(par[node]);
}else{
return node;

}


}
void unit(int a,int b){

int aa=fin(a);
int bb=fin(b);
if(siz[a]>siz[b]){
    swap(a,b);
}

if(aa==bb){
    return;
}
par[aa]=bb;
//ans+=(mx[aa]+mx[bb]);
mx[bb]=max(mx[bb],mx[aa]);
return;



 }
 int active[mxx];
int main(){




cin>>n;
pair<int,int>v[n];
for(int i=0;i<n;i++){
    cin>>a[i];
    v[i]={a[i],i};
}
 for(int i=0;i<n-1;i++){
    int x,y;
    cin>>x>>y;
    x--;y--;
    adj[x].push_back(y);
    adj[y].push_back(x);
    edge[i]={x,y};

 }
 init();
 sort(v,v+n);

vector<pair<int,int>>x;
 for(int i=0;i<n;i++){

    int me=v[i].second;
    active[me]=1;
    for(auto it:adj[me]){

        if(active[it]){
               ans+=v[i].first;
               ans+=mx[it];
        }


    }

    for(auto it:adj[me]){

        if(active[it]){
              unit(it,me);
        }


    }

}





ll ans2=inf;


int c=-1;

 cout<<ans;

}

Compilation message (stderr)

sjekira.cpp: In function 'int main()':
sjekira.cpp:109:4: warning: unused variable 'ans2' [-Wunused-variable]
  109 | ll ans2=inf;
      |    ^~~~
sjekira.cpp:112:5: warning: unused variable 'c' [-Wunused-variable]
  112 | int c=-1;
      |     ^

• Subtask 1: 0 / 10
• Subtask 2: 0 / 10
• Subtask 3: 0 / 30
• Subtask 4: 0 / 60