POJ 3767 I Wanna Go Home

不错的题目，由于只能从国度1到国度2一次，我原来想建两张图，一张都是国度1的点，另一张都是国度2的点，分别以城市1和城市2为源点求最短路径再枚举，后来看了别人的（承认这点不好，自己很少深入思考），只要一张图就行了，做法就是把2到1的边换成单向的，只能从1到2（因为一开始在1）

 

自己能够想到一些东西，但是缺乏有效利用这些东西的能力，比如这道题的建图方法，很巧妙，虽然基本的性质我知道，但就是想不到，可能是对题目意思和图的理解的还不够深入吧

 

代码：

#include<iostream>using namespace std;#define inf 999999int g[605][605];int dis[605],u[605],c[605];void dijkstra(int source,int n){int i,j,k,min;memset(u,0,sizeof(u));for(i=1;i<=n;i++)dis[i]=inf;dis[source]=0;for(i=1;i<=n;i++){min=inf;for(j=1;j<=n;j++){if(!u[j]&&dis[j]<min){min=dis[j];k=j;}}u[k]=1;for(j=1;j<=n;j++){if(g[k][j])//如果存在边{if(g[k][j]+dis[k]<dis[j]){dis[j]=g[k][j]+dis[k];}}}}}int main(){int i,j,k,n,m,a,b,t,w;while(scanf("%d",&n)!=EOF&&n){memset(g,0,sizeof(g));memset(c,0,sizeof(c));scanf("%d",&m);for(i=1;i<=m;i++){scanf("%d %d %d",&a,&b,&w);g[a][b]=g[b][a]=w;}for(i=1;i<=n;i++)scanf("%d",&c[i]);for(i=1;i<=n;i++){if(c[i]==2){for(j=1;j<=n;j++){if(c[j]==1)g[i][j]=0;//删除2到1的边}}}dijkstra(1,n);if(dis[2]==inf){printf("-1/n");}elseprintf("%d/n",dis[2]);}return 0;}