nysit 203 最短路+dp

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=203

题目大意：有n+1座城池，小白在第0座城池里，其他城池从1到n编号并且每座城池拥有一些财富，这些城池间有m条道路，现在小白想要拥有这些财富，需要攻打占有就可以了，但是派武将攻打时，需要消耗粮食，但粮食有限，问小白最多能够拥有的财富

思路：先求各点到0的最短路，然后用01背包问题求最大财富

#include<stdio.h> #include<string.h> #define MAX 999999999 #define N 110 int map[N][N],s[N],dis[N],v[N]; int dp[1000010]; int max(int a,int b) { return a>b?a:b; } int main() { int ncase,i,j,k,x,y,w,n,m,S,u,min; scanf("%d",&ncase); while(ncase--) { scanf("%d%d%d",&S,&n,&m); for(i=0;i<=n;i++) { /*初始化*/ s[i]=0; dis[i]=MAX; for(j=0;j<=n;j++) map[i][j]=MAX; map[i][i]=0; } for(i=1;i<=m;i++) { scanf("%d%d%d",&x,&y,&w); if(map[x][y]>w) /*两个城池间可能有多条道路*/ map[x][y]=map[y][x]=w; } for(i=1;i<=n;i++) scanf("%d",&v[i]); s[0]=1; /*下面的是dijkstra算法求最短路*/ dis[0]=0; u=0; for(i=1;i<=n;i++) { for(j=1;j<=n;j++) /*修改最短路*/ if(s[j]==0 && map[u][j]!=MAX && dis[j]>dis[u]+map[u][j]) dis[j]=dis[u]+map[u][j]; min=MAX; for(j=1;j<=n;j++) /*加入最近的顶点*/ if(s[j]==0 && dis[j]<min) { min=dis[j]; u=j; } s[u]=1; } memset(dp,0,sizeof(dp)); /*此处第一次写成sizeof(dp[0])了，wrong了一次*/ for(i=1;i<=n;i++) /*01背包问题*/ for(k=S;k>=dis[i];k--) dp[k]=max(dp[k],dp[k-dis[i]]+v[i]); printf("%d/n",dp[S]); } return 0; }