[HNOI2006]公路修建问题

题目描述

输入输出格式

输入格式：

在实际评测时，将只会有m-1行公路

输出格式：

输入输出样例

输入样例#1： 复制

4 2 5
1 2 6 5
1 3 3 1
2 3 9 4
2 4 6 1
3 4 4 2

输出样例#1： 复制

4
2 1
3 2
5 1

输入样例#2： 复制

4 1 5
1 2 6 5
1 3 3 1
2 3 9 4
2 4 6 1
3 4 4 3

输出样例#2： 复制

3
2 1
4 2
5 2
贪心，先搞出k条一级路径
把之后没选的边按min(c1,c2)排序
水题

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node
{
    int u,v,d1,d2,pd,id;
}ed[200001];
int ans,n,k,m,num,set[100001];
bool cmp1(Node a,Node b)
{
    return a.d1<b.d1;
}
bool cmp2(Node a,Node b)
{
    return min(a.d1,a.d2)<min(b.d1,b.d2);
}
bool cmp3(Node a,Node b)
{
    return a.id<b.id;
}
int find(int x)
{
    if (set[x]!=x) set[x]=find(set[x]);
    return set[x];
}
int main()
{int i,j,u,v,c1,c2,cnt,st;
     cin>>n>>k>>m;
     m--;
      for (i=1;i<=m;i++)
      {
          scanf("%d%d%d%d",&u,&v,&c1,&c2); 
          ed[++num].u=u,ed[num].v=v,ed[num].d1=c1; ed[num].d2=c2;
          ed[num].id=i;
      }
      sort(ed+1,ed+num+1,cmp1);
      cnt=0;
      for (i=1;i<=n;i++)
      set[i]=i;
      for (i=1;i<=num;i++)
      {
           int p=find(ed[i].u),q=find(ed[i].v);
           if (p!=q)
           {
               ans=max(ans,ed[i].d1);
               cnt++;
               ed[i].pd=1;
               set[p]=q;
              if (cnt==k) break;
           }
      }
      st=i;
      sort(ed+1,ed+num+1,cmp2);
      for (i=1;i<=num;i++)
      if (ed[i].pd==0)
    {
          int p=find(ed[i].u),q=find(ed[i].v);
           if (p!=q)
           {
               ans=max(ans,min(ed[i].d1,ed[i].d2));
               cnt++;
               set[p]=q;
               ed[i].pd=2;
              if (cnt==n-1) break;
           }    
      }
      cout<<ans<<endl;
      sort(ed+1,ed+num+1,cmp3);
      for (i=1;i<=num;i++)
      if (ed[i].pd)
      {
          printf("%d %d
",ed[i].id,ed[i].pd);
      }
}