比赛 |
EYOI常规赛 1st |
评测结果 |
WWWWWWWWWW |
题目名称 |
树染色 |
最终得分 |
0 |
用户昵称 |
今天作业写了没 |
运行时间 |
0.055 s |
代码语言 |
C++ |
内存使用 |
8.86 MiB |
提交时间 |
2021-12-02 21:09:10 |
显示代码纯文本
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,s,s1,s2,ans;
struct in
{
ll so[1005],m,l;
}p[1005],x;
struct cmp
{
bool operator ()(in a,in b)
{
return a.m<b.m;
}
};
priority_queue<in,vector<in>,cmp> q;
inline ll read()
{
ll x=0;
char a;
a=getchar();
while(a<'0'||a>'9') a=getchar();
while(a>='0'&&a<='9')
{
x=(x<<3)+(x<<1)+(a^48);
a=getchar();
}
return x;
}
ll write(ll x)
{
if(x/10) write(x/10);
putchar(x-x/10*10+48);
}
void in()
{
n=read();
s=read();
for(ll i=1;i<=n;i++)
{
p[i].m=read();
}
for(ll i=1;i<n;i++)
{
s1=read();
s2=read();
p[s1].so[p[s1].l++]=s2;
}
q.push(p[s]);
}
ll deal(ll t)
{
x=q.top();
ans+=x.m*t;
q.pop();
for(int i=0;i<x.l;i++)
{
q.push(p[x.so[i]]);
}
if(q.empty()) return ans;
deal(++t);
}
int main()
{
freopen("colortree.in","r",stdin);
freopen("colortree.out","w",stdout);
in();
write(deal(1));
return 0;
}