题目地址:UESTC 1220
比赛的时候翻译完想了一小会就没再管。结果K题也没调试出来。。
这题是很神奇的图论题,建图方式是从y[i]到x[i]连一条有向边,权值为c[i]。然后将所有重要性为0的设为源点,然后跑最短路,结果就是所有重要性为2的点的最短距离。
实在是不好解释和证明这种建图的正确性。。有种只可意会不可言传的感觉。。需要仔细思考。
代码如下:
#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
#include <string>
#include <time.h>
using namespace std;
#define LL long long
#define pi acos(-1.0)
#pragma comment(linker, "/STACK:1024000000")
const int mod=9901;
const LL INF=(LL)1e13;
const double eqs=1e-9;
const int MAXN=100000+10;
LL d[MAXN];
int head[MAXN], cnt, c[MAXN], x[MAXN], y[MAXN], w[MAXN], vis[MAXN];
struct node
{
int u, v, next;
LL w;
}edge[MAXN];
queue<int>q;
void add(int u, int v, LL w)
{
edge[cnt].v=v;
edge[cnt].w=w;
edge[cnt].next=head[u];
head[u]=cnt++;
}
void init(int m)
{
memset(head,-1,sizeof(head));
cnt=0;
for(int i=1;i<=m;i++){
d[i]=INF;
}
while(!q.empty()) q.pop();
memset(vis,0,sizeof(vis));
}
void spfa()
{
while(!q.empty()){
int u=q.front();
q.pop();
vis[u]=0;
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].v;
if(d[v]>d[u]+edge[i].w){
d[v]=d[u]+edge[i].w;
if(!vis[v]){
q.push(v);
vis[v]=1;
}
}
}
}
}
int main()
{
int t, n, m, i, j, flag, icase=0;
LL ans;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
init(m);
for(i=1;i<=n;i++){
scanf("%d",&x[i]);
}
for(i=1;i<=n;i++){
scanf("%d",&y[i]);
}
for(i=1;i<=n;i++){
scanf("%d",&c[i]);
add(y[i],x[i],(LL)c[i]);
}
for(i=1;i<=m;i++){
scanf("%d",&w[i]);
if(w[i]==0) {
q.push(i);
d[i]=0;
vis[i]=1;
}
}
spfa();
ans=0;
flag=0;
for(i=1;i<=m;i++){
if(w[i]==2){
if(d[i]==INF) {
flag=1;
break;
}
ans+=d[i];
}
}
printf("Case #%d: ",++icase);
if(flag) puts("-1");
else printf("%lld\n",ans);
}
return 0;
}