nyoj 118 修路方案

http://acm.nyist.net/JudgeOnline/problem.php?pid=118

这道题，不想用想啊。次小生成树，裸裸的！
其实这个代码是有问题的，这个数据不够严谨，比如，图的输入，应该是需要考虑重边的，我这个代码没有考虑重边，但是AC了，是不够完美的！
我这个是使用类似于dijskal的思想，其实使用O(n^2)的prim算法效果是一样的，并不会优化太多，因为你自己这句话。

for(i=1;i<=N;i++)
            if(vis[i]&&i!=k){ maxlen[i][k]=maxlen[k][i]=max(maxlen[i][pr],maxlen[k][pr]); // cout<<"i="<<i<<" k="<<k<<" "<<maxlen[i][k]<<endl; }

这个就已经觉定了你这个算法的复杂度是平方了，不会更好了！
所以使用dijskal优化并没有什么必要，我相信也不会有题目卡这个复杂度。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include<cstring>
#include<queue>
using namespace std;
#include<vector>
typedef struct notes{
    int c,p;
    bool operator > (const notes& aa)const{
        return c>aa.c;
    }
}note;
#define maxN 510
#define MAX 0x3f3f3f3f
#define MIN -0x3f3f3f3f
int N,M,map[maxN][maxN],dis[maxN],maxlen[maxN][maxN],pre[maxN];
bool mst[maxN][maxN];
bool vis[maxN];
int mdis[maxN];
int prim()
{
    int i,j,k,minn,pr;
    memset(vis,false,sizeof(vis));
    memset(mst,false,sizeof(mst));
    priority_queue<note,vector<note>,greater<note> >  que;
    while(!que.empty()){
        que.pop();
    }
    note p1;
    for(i=1;i<=N;i++){

        p1.p=i;
        p1.c=map[i][1];
        que.push(p1);
        pre[i]=1;
        mdis[i]=map[i][1];
    }
    vis[1]=true;
    int ans=0;
    while(!que.empty()){
        note temp=que.top();
        que.pop();
        k=temp.p;
        if(vis[k])continue;
    // printf("vis=%d\n",k);
        pr=pre[k]; 
    // printf("k=%d pr=%d %d\n",k,pr,map[k][pr]);
        maxlen[k][pr]=maxlen[pr][k]=map[k][pr];

        for(i=1;i<=N;i++)
            if(vis[i]&&i!=k){

                maxlen[i][k]=maxlen[k][i]=max(maxlen[i][pr],maxlen[k][pr]);
        // cout<<"i="<<i<<" k="<<k<<" "<<maxlen[i][k]<<endl;
            }

        vis[k]=true;
    // cout<<"temp.c="<<temp.c<<endl;
        ans+=temp.c;
        mst[k][pr]=mst[pr][k]=true;
        for(i=1;i<=N;i++)
            if(!vis[i]&&(mdis[i]>map[i][k]))
            {
                mdis[i]=map[i][k];
                p1.p=i;
                p1.c=map[i][k];
                que.push(p1);
                pre[i]=k;
            }
    }
// cout<<"ans="<<ans<<endl;

    int min_=99999999;
    for(i=1;i<N;i++)
        for(j=i+1;j<=N;j++)
            {
            // if(i==1&&j==4){
            // printf("min_=%d\n",min_);
            // }
            if((!mst[i][j])&&map[i][j]!=MAX){
            // cout<<"min_="<<min_<<endl;
                 min_=min(min_,ans+map[i][j]-maxlen[i][j]);
            // cout<<"min_==="<<min_<<endl;
                }
            // if(i==1&&j==4){
            // printf("min_=%d\n",min_);
            // }
            }
    if(min_==ans)return 1;
    return 0;
}
int main()
{
    //freopen("1.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int u,v,w;
        scanf("%d%d",&N,&M);
        for(int i=0;i<=N;i++)
            for(int j=0;j<=N;j++)
            {
                map[i][j]=MAX;
                maxlen[i][j]=MIN;
            }
        for(int i=0;i<M;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            map[u][v]=map[v][u]=w;
        }
        if(prim())printf("Yes\n");
        else printf("No\n");
    }
}