poj 1639 Picnic Planning

有度限制的最小生成树。解法网上很多，就不再叙述了。

#include <iostream>#include <cstdio>#include <cstring>using namespace std;const int maxn=1e2+9,inf=5e7;char a[maxn][maxn];int n,m,limit;int head[maxn],lon,visit[maxn],dist[maxn],distfrom[maxn];bool flag[maxn];struct{    int to,next,w,use;}e[maxn*maxn<<1];void edgeini(){    memset(head,-1,sizeof(head));    lon=-1;}void edgemake(int from,int to,int w){    e[++lon].to=to;    e[lon].w=w;    e[lon].next=head[from];    e[lon].use=0;    head[from]=lon;}int get(int &t){    for(int i=1;i<t;i++)    if(strcmp(&a[t][1],&a[i][1])==0)    {        t--;        return i;    }    return t;}int prim(int t){    int key[maxn],ff[maxn],from[maxn];    memset(key,50,sizeof(key));    memset(ff,0,sizeof(ff));    key[t]=0;    ff[t]=1;    visit[t]=t;    for(int k=head[t];k!=-1;k=e[k].next)    {        int u=e[k].to;        if(key[u]>e[k].w)        {            key[u]=e[k].w;            from[u]=k;        }    }    int ans=0;    while(1)    {        int tmp=inf,now;        for(int i=1;i<=n;i++)        if(key[i]<tmp&&!ff[i]&&!visit[i])        {            tmp=key[i];            now=i;        }        if(tmp==inf) return ans;        ans+=tmp;        ff[now]=1;        visit[now]=t;        e[from[now]].use=1;        e[from[now]^1].use=1;        for(int k=head[now];k!=-1;k=e[k].next)        {            int u=e[k].to;            if(key[u]>e[k].w)            {                key[u]=e[k].w;                from[u]=k;            }        }    }}void dfs(int t,int from,int mm,int kk){    for(int k=head[t];k!=-1;k=e[k].next)    {        int u=e[k].to;        if((!e[k].use)||(u==from)) continue;        if(e[k].w>mm)        {            dist[u]=e[k].w;            distfrom[u]=k;            dfs(u,t,e[k].w,k);        }        else        {            dist[u]=mm;            distfrom[u]=kk;            dfs(u,t,mm,kk);        }    }}int prim(){    int ans=0;    memset(visit,0,sizeof(visit));    memset(flag,0,sizeof(flag));    visit[m]=m;    int sum=0;    for(int i=1;i<=n;i++)    if(!visit[i])    {        ans+=prim(i);        sum++;    }    for(int i=1;i<=sum;i++)    {        int tmp=inf,now;        for(int k=head[m];k!=-1;k=e[k].next)        {            if(e[k].w<tmp&&!flag[visit[e[k].to]])            {                tmp=e[k].w;                now=k;            }        }        flag[visit[e[now].to]]=1;        e[now].use=1;        e[now^1].use=1;        ans+=tmp;    }    for(int i=sum+1;i<=limit;i++)    {        memset(dist,50,sizeof(dist));        dfs(m,-1,0,-0);        int tmp=0,now,nowk;        for(int k=head[m];k!=-1;k=e[k].next)        {            if(e[k].use) continue;            if(dist[e[k].to]-e[k].w>tmp)            {                tmp=dist[e[k].to]-e[k].w;                now=k;                nowk=distfrom[e[k].to];            }        }        if(tmp==0) return ans;        else        {            e[now].use=1;            e[now^1].use=1;            e[nowk^1].use=0;            e[nowk].use=0;            ans-=tmp;        }    }    return ans;}int main(){//    freopen("in.txt","r",stdin);    while(scanf("%d",&n)!=EOF)    {        memset(a,0,sizeof(a));        edgeini();        int mm;        for(int i=1,k=0,w;i<=n;i++)        {            scanf("%s",&a[++k][1]);            int t=get(k);            scanf("%s",&a[++k][1]);            scanf("%d",&w);            int s=get(k);            edgemake(t,s,w);            edgemake(s,t,w);            mm=k;        }        n=mm;        scanf("%d",&limit);        for(m=1;;m++) if(strcmp(&a[m][1],"Park")==0) break;        printf("Total miles driven: ");        cout<<prim()<<endl;    }    return 0;}