比赛 |
20161115 |
评测结果 |
AWWWAAWWWWWWWWWWWWWW |
题目名称 |
树和机器人 |
最终得分 |
15 |
用户昵称 |
coolkid |
运行时间 |
0.128 s |
代码语言 |
C++ |
内存使用 |
5.65 MiB |
提交时间 |
2016-11-15 11:56:09 |
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=1e5+10;
const int MAXM=MAXN<<1;
struct Edge{
int from,to,dist,nxt;
}edges[MAXM];
int head[MAXN],cnt=0;
void addedge(int f,int t,int d){
edges[cnt].from=f;edges[cnt].to=t;edges[cnt].dist=d;
edges[cnt].nxt=head[f];
head[f]=cnt++;
}
int n,r,k;
int dist[MAXM];
int son[MAXN];
long long ans=0;
void init(){
memset(head,-1,sizeof(head));
scanf("%d%d%d",&n,&r,&k);
for(int i=1;i<n;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
addedge(x,y,z);
addedge(y,x,z);
ans+=z;
}
cout<<ans<<endl;
exit(0);
memset(son,0,sizeof(son));
}
int deep[MAXN];
int fa[MAXN];
int dp=0;
void dfs(int u){
for(int i=head[u];~i;i=edges[i].nxt){
int v=edges[i].to;
if(fa[u]==v) continue;
fa[v]=u;
deep[v]=deep[u]+1;
dp=max(dp,deep[v]);
dfs(v);
}
}
int sz=0;
void work(){
dfs(r);
long long ans1=0;
for(int i=1;i<=n;i++) if(deep[i]==dp){
for(int j=head[i];~j;j=edges[j].nxt) dist[sz++]=edges[j].dist;
}
sort(dist,dist+sz);
ans1=ans;
if(k==1) ans1=ans*2-dist[sz-1];
cout<<ans1<<endl;
}
int main(){
freopen("trobot.in","r",stdin);
freopen("trobot.out","w",stdout);
init();
work();
return 0;
}