【USACO題庫】3.2.6 Sweet Butter香甜的黃油
阿新 • • 發佈:2018-12-16
這一題,我本來是想用spfa演算法的,但後來嫌太累了,就打了個習以為常的floyd演算法。
floyd演算法很簡單——三重迴圈。。。
for (k=1;i<=p;i++)
for (i=1;j<=p;j++)
for (j=i+1;j<=p;j++)
{
...
}
後來,發現時間有些慢了(雖然也能過),但依舊改回了spfa。
先來看一看floyd的:
#include<cstdio>
#include<cstring>
#define dislike 666666666
using namespace std;
int b[501],f[801][801];
int n,p,c;
int dg(int x)
{
int s=0;
for (int i=1;i<=n;i++)
if (x!=b[i]) s+=f[b[i]][x];
return s;
}
int main()
{
freopen("butter.in","r",stdin);
freopen("butter.out","w",stdout);
int i,j,k,x,ans;
scanf("%d %d%d",&n,&p,&c);
for (i=1;i<=p;i++)
for (j=1;j<=p;j++)
f[i][j]=dislike;
for (i=1;i<=n;i++)
{
scanf("%d",&b[i]);
}
for (i=1;i<=c;i++)
{
scanf("%d%d%d",&j,&k,&x);
f[j][k]=x;
f[k][j]=x;
}
for (k=1;k<=p;k++)
{
for (i=1;i<=p;i++)
{
for (j=i+1;j<=p;j++)
{
if (f[i][k]+f[k][j]<f[i][j])
{
f[i][j]=f[i][k]+f[k][j];
f[j][i]=f[i][j];
}
}
}
}
ans=999999999;
for (i=1;i<=p;i++)
{
x=dg(i);
if (x<ans) ans=x;
}
printf("%d\n",ans);
return 0;
}
再來看一看spfa的:
#include<cstdio>
#include<cstring>
using namespace std;
int f[510];
struct node
{
int x,y,c,next;
}a[11000];int len,last[810];
void ins(int x,int y,int c)
{
len++;
a[len].x=x;a[len].y=y;a[len].c=c;
a[len].next=last[x];last[x]=len;
}
int mymin(int x,int y) {return x<y?x:y;}
int n,p,c;
int d[810],st,ed;
bool v[810];
int list[810],head,tail;
void spfa()
{
for(int i=1;i<=p;i++)
{
d[i]=999999999;
}
d[st]=0;
memset(v,false,sizeof(v));
v[st]=true;
list[1]=st;head=1;tail=2;
if(tail==p+1) tail=1;
while(head!=tail)
{
int x=list[head];
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(d[y]>d[x]+a[k].c)
{
d[y]=d[x]+a[k].c;
if(v[y]==false)
{
v[y]=true;
list[tail]=y;
tail++;if(tail==p+1) tail=1;
}
}
}
v[x]=false;
head++;
if(head==p+1) head=1;
}
}
int main()
{
scanf("%d%d%d",&n,&p,&c);
len=0;memset(last,0,sizeof(last));
for(int i=1;i<=n;i++)
{
scanf("%d",&f[i]);
}
for(int i=1;i<=c;i++)
{
int x,y,c;
scanf("%d%d%d",&x,&y,&c);
ins(x,y,c);
ins(y,x,c);
}
int min=999999999;
for(int i=1;i<=p;i++)
{
int ans=0;
st=i;
spfa();
for(int j=1;j<=n;j++)
{
if(d[f[j]]==999999999)
{
ans=999999999;
break;
}
ans+=d[f[j]];
}
min=mymin(min,ans);
}
printf("%d\n",min);
return 0;
}
讓咱們看一看時間變化
從floyd:
再到spfa:
雖然在程式碼長度上有一點差別,但是時間!!!!!
782ms——————>74ms,我*!
算了,不說了,希望此篇對讀者有些幫助!