luogu P1340 兽径管理 WA

题目描述

约翰农场的牛群希望能够在 N 个(1<=N<=200) 草地之间任意移动。草地的编号由 1到 N。草地之间有树林隔开。牛群希望能够选择草地间的路径，使牛群能够从任一片草地移动到任一片其它草地。牛群可在路径上双向通行。

牛群并不能创造路径，但是他们会保有及利用已经发现的野兽所走出来的路径(以下简称兽径)。每星期他们会选择并管理一些或全部已知的兽径当作通路。

牛群每星期初会发现一条新的兽径。他们接着必须决定管理哪些兽径来组成该周牛群移动的通路，使得牛群得以从任一草地移动到任一草地。牛群只能使用当周有被管理的兽径做为通路。

牛群希望他们管理的兽径长度和为最小。牛群可以从所有他们知道的所有兽径中挑选出一些来管理。牛群可以挑选的兽径与它之前是否曾被管理无关。

兽径决不会是直线，因此连接两片草地之间的不同兽径长度可以不同。此外虽然两条兽径或许会相交，但牛群非常的专注，除非交点是在草地内，否则不会在交点换到另外一条兽径上。

在每周开始的时候，牛群会描述他们新发现的兽径。如果可能的话，请找出可从任何一草地通达另一草地的一组需管理的兽径，使其兽径长度和最小。

输入输出格式

输入格式：

输入的第一行包含两个用空白分开的整数 N 和 W。W 代表你的程序需要处理的周数. (1 <= W <= 6000)。

以下每处理一周，读入一行数据，代表该周新发现的兽径，由三个以空白分开的整数分别代表该兽径的两个端点 (两片草地的编号) 与该兽径的长度(1…10000)。一条兽径的两个端点一定不同。

输出格式：

每次读入新发现的兽径后，你的程序必须立刻输出一组兽径的长度和，此组兽径可从任何一草地通达另一草地，并使兽径长度和最小。如果不能找到一组可从任一草地通达另一草地的兽径，则输出 “-1”。

输入输出样例

输入样例#1：

输出样例#1：

-1 //No trail connects 4 to the rest of the fields.
-1 //No trail connects 4 to the rest of the fields.
-1 //No trail connects 4 to the rest of the fields.
14 //Maintain 1 4 3, 1 3 8, and 3 2 3.
12 //Maintain 1 4 3, 1 3 6, and 3 2 3.
8 //Maintain 1 4 3, 2 1 2, and 3 2 3.
//program exit

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;
const int N=2001;

int f[N];
struct node{
    int u,v,w;
}E[N*2];
int n,when;

inline int read()
{
    int x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
    return x;
}

int getfa(int x)
{    return f[x]==x?x:f[x]=getfa(f[x]);    }

bool cmp(node a,node b)
{    return a.w<b.w;    }

int main()
{
    n=read();
    when=read();
    
    for(int i=1;i<=when;i++)
    {
        E[i].u=read();
        E[i].v=read();
        E[i].w=read();
        if(i<n-1)
        {
            printf("-1
");
            continue;
        }
        else
        {
            int tot=0;
            int answer=0;
            bool flag=1;
            sort(E+1,E+i+1,cmp);
            for(int j=1;j<=n;j++)
                f[j]=j;
            for(int j=1;j<=i&&flag;j++)
            {
                int fu=getfa(E[j].u);
                int fv=getfa(E[j].v);
                if(fu!=fv)
                {
                    f[fu]=fv;
                    tot++;
                    answer+=E[j].w;
                }
                if(tot==n-1)
                    flag=0;
            }
            if(tot!=n-1)
                printf("-1
");
            else
                printf("%d
",answer);
        }
    }
    return 0;
}
/*
4 6
1 2 10
1 3 8
3 2 3
1 4 3
1 3 6
2 1 2
*/