| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 6606 | mrcamel | 비용 (KOI11_cost) | C++98 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <algorithm>
using namespace std;
#define in cin
#define out cout
#define DIV 1000000000
#define int64 long long int
struct LL
{
int s, e, w;
}L[100001];
struct VV
{
int subN;
int parent;
}Vtx[100001];
int cmp(const LL &a, const LL &b)
{
return a.w > b.w;
}
int64 sum[100001]; //합
int main()
{
int n, k;
in >> n >> k; //정점, 간선
//scanf("%d %d", &n, &k);
for(int i=0; i<k; i++)
//scanf("%d %d %d", &L[i].s, &L[i].e, &L[i].w);
in >> L[i].s >> L[i].e >> L[i].w;
sort(L, L+k, cmp);
for(int i=1; i<=n; i++)
{
Vtx[i].parent = i;
Vtx[i].subN = 1;
}
for(int i=0; i<k; i++) {
if(i != 0) sum[i] = sum[i-1] + L[k-i-1].w;
else sum[i] = L[k-1].w;
} reverse(sum, sum+k);
int64 res = 0;
for(int i=0; i<k; i++)
{
int v1 = L[i].s;
int v2 = L[i].e;
int p1 = v1, p2 = v2;
while( p1 != Vtx[p1].parent ) p1 = Vtx[p1].parent;
while( p2 != Vtx[p2].parent ) p2 = Vtx[p2].parent;
if(p1 != p2)
{
res += (Vtx[p1].subN * Vtx[p2].subN * sum[i]);
res %= DIV;
Vtx[p1].subN += Vtx[p2].subN;
Vtx[p2].parent = p1;
}
}
printf("%d", res);
return 0;
}