[kuangbin带你飞]专题四 最短路练习 O LightOJ 1074

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=205;
int a[maxn];
vector<pair<int,int> >E[maxn];
int d[maxn];
bool vis[maxn];
int in[maxn];
int n,m,q;
void init()
{
for(int i=0;i<maxn;i++)
{
E[i].clear();
d[i]=INF;
a[i]=0;
vis[i]=false;
in[i]=0;
}
}

void spfa(int s)
{
queue<int>q;
vis[s]=true;
d[s]=0;
q.push(s);
while(!q.empty())
{
int now=q.front();
q.pop();
if(in[now]++>n)
return;
vis[now]=false;
for(int i=0;i<E[now].size();i++)
{
int v=E[now][i].first;
if(d[v]>d[now]+E[now][i].second)
{
d[v]=d[now]+E[now][i].second;
if(vis[v])
continue;
vis[v]=true;
q.push(v);
}
}

}
}

int main()
{
int t;
scanf("%d",&t);
for(int casei=1;casei<=t;casei++)
{
init();
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
scanf("%d",&m);
for(int i=0;i<m;i++)
{
int c,b;
scanf("%d%d",&c,&b);
int temp=(a[b]-a[c])*(a[b]-a[c])*(a[b]-a[c]);
E[c].push_back(make_pair(b,temp));
}
spfa(1);
scanf("%d",&q);
printf("Case %d:\n",casei);
while(q--)
{
int temp;
scanf("%d",&temp);
if(d[temp]==INF || d[temp]<3)
printf("?\n");
else
printf("%d\n",d[temp]);
}
}
}