【51nod 1737 配对】+ 链式前向星 + dfs + 输入外挂

1737 配对
基准时间限制：1 秒 空间限制：131072 KB 分值: 40 难度：4级算法题
收藏
关注
给出一棵n个点的树，将这n个点两两配对，求所有可行的方案中配对两点间的距离的总和最大为多少。
Input
一个数n（1<=n<=100,000,n保证为偶数）
接下来n-1行每行三个数x,y,z表示有一条长度为z的边连接x和y（0<=z<=1,000,000,000）
Output
一个数表示答案
Input示例
6
1 2 1
1 3 1
1 4 1
3 5 1
4 6 1
Output示例
7
//配对方案为（1，2）（3，4）（5，6）

一条边的贡献 = 点较少的一边的点的个数 * 这条边的权值

AC代码： （453 ms）

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int K = 1e5 + 10;
typedef long long LL;
LL n,head[K],s[K],num,ans;
struct node{
    LL to,vl,next;
}st[K * 2];
void add(LL a,LL b,LL c){
    st[num] = node{b,c,head[a]}; head[a] = num++;
    st[num] = node{a,c,head[b]}; head[b] = num++;
}
void dfs(LL x,LL f){
    s[x] = 1;
    for(int i = head[x]; i != -1; i = st[i].next)
        if(st[i].to != f)
            dfs(st[i].to,x),s[x] += s[st[i].to],ans += min(s[st[i].to],n - s[st[i].to]) * st[i].vl;
}
int main()
{
    LL a,b,c;
    num = 0;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i = 1; i < n; i++)
         scanf("%lld %lld %lld",&a,&b,&c),add(a,b,c);
    ans = 0;
    dfs(1,0);
    printf("%lld\n",ans);
    return 0;
}

AC代码 + 输入外挂 （171ms）：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int K = 1e5 + 10;
typedef long long LL;
LL n,head[K],s[K],num,ans;
struct node{
    LL to,vl,next;
}st[K * 2];
LL read(){
    LL x = 0, y = 1;
    char s = getchar();
    while(s < '0' || s > '9') { if(s == '-') y = -1;s = getchar();}
    while(s >= '0' && s <= '9') x = x * 10 + s - '0',s = getchar();
    return x * y;
}
void add(LL a,LL b,LL c){
    st[num] = node{b,c,head[a]}; head[a] = num++;
    st[num] = node{a,c,head[b]}; head[b] = num++;
}
void dfs(LL x,LL f){
    s[x] = 1;
    for(int i = head[x]; i != -1; i = st[i].next)
        if(st[i].to != f)
            dfs(st[i].to,x),s[x] += s[st[i].to],ans += min(s[st[i].to],n - s[st[i].to]) * st[i].vl;
}
void out(LL a){
   if(a < 0) putchar('-'), a = -a;
   if(a >= 10) out(a / 10); putchar(a % 10 + '0');
}
int main()
{
    LL a,b,c;
    num = 0;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i = 1; i < n; i++)
         a = read(),b = read(),c = read(),add(a,b,c);
    ans = 0;
    dfs(1,0);
    out(ans);
    return 0;
}