Description
Vasya在玩一个叫做”Dwarf Tower”的游戏,这个游戏中有n个不同的物品,它们的编号为1到n。现在Vasya想得到编号为1的物品。
获得一个物品有两种方式:
1. 直接购买该物品,第i件物品花费的钱为ci
2. 用两件其他物品合成所需的物品,一共有m种合成方式。
请帮助Vasya用最少的钱获得编号为1的物品。
Input
第一行有两个整数n,m(1<=n<=10000,0<=m<=100000),分别表示有n种物品以及m种合成方式。
接下来一行有n个整数,第i个整数ci表示第i个物品的购买价格,其中0<=ci<=10^9。
接下来m行,每行3个整数ai,xi,yi,表示用物品xi和yi可以合成物品ai,其中(1<=ai,xi,yi<=n; ai<>xi, xi<>yi, yi<>ai)
Output
一行,一个整数表示获取物品1的最少花费。
Sample Input
5 3
5 0 1 2 5
5 2 3
4 2 3
1 4 5
Sample Output
2
Data Constraint
60%的数据,n<=100
100%的数据,n<=10000,m<=100000
思路:
一开始我觉得是树形DP
但又想了想,合成关系不保证a,b < c,且不保证合成关系不形成环。
于是,我便思路一转,想到了spfa。
把每个合成关系连一条边 ,权值为另一个点的价值,再用一个原点,给所有点连一条边,就等于最短路。
代码:
#include<cstdio>
#include<queue>
using namespace std;
int n,m;
struct node
{
long long p,next,f;
}s[200002];
long long a[10001],c=0,b[10001],x,y,z;
bool f[10001]={0};
queue<int> d;
void add(int x,int y,int z)
{
c++;
s[c].p=y;
s[c].next=b[x];
b[x]=c;
s[c].f=z;
}
void bfs(int x)
{
long long xx;
for(int i=1;i<=n;i++)
{
d.push(i);
f[i]=1;
}
while(!d.empty())
{
xx=d.front();
f[xx]=0;
d.pop();
for(int i=b[xx];i;i=s[i].next)
if(a[xx]+a[s[i].p]<a[s[i].f])
{
a[s[i].f]=a[xx]+a[s[i].p];
if(!f[s[i].f])
{
d.push(s[i].f);
f[s[i].f]=1;
}
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&z,&x,&y);
add(x,y,z);
add(y,x,z);
}
bfs(1);
printf("%d\n",a[1]);
return 0;
}