PAT (Advanced Level) 1072. Gas Station (30)

枚举一下选的位置，每次算一下就可以了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<stack>
#include<queue>
#include<string>
#include<algorithm>
using namespace std;

const int INF=0x7FFFFFFF;
const int maxn=+;
int n,m,k,ds;
struct Edge
{
    int u,v;
    int dis;
}e[*maxn];
vector<int>g[maxn];
int tot;
int flag[maxn],dis[maxn];
double ans1,ans2;
int ans;
int MIN;

int get(char *s)
{
    int num=;
    if(s[]!='G')
    {
        for(int i=;s[i];i++)
            num=num*+s[i]-'';
    }
    else
    {
        for(int i=;s[i];i++)
            num=num*+s[i]-'';
        num=num+n;
    }
    return num;
}

void SPFA(int s)
{
    queue<int>Q;
    memset(flag,,sizeof flag);
    for(int i=;i<=n+m;i++) dis[i]=INF;
    Q.push(s); flag[s]=; dis[s]=;
    while(!Q.empty())
    {
        int head=Q.front(); Q.pop(); flag[head]=;
        for(int i=;i<g[head].size();i++)
        {
            int id=g[head][i];
            if(dis[head]+e[id].dis<dis[e[id].v])
            {
                dis[e[id].v]=dis[head]+e[id].dis;
                if(flag[e[id].v]==)
                {
                    flag[e[id].v]=;
                    Q.push(e[id].v);
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&k,&ds);

    tot=;
    for(int i=;i<=k;i++)
    {
        char u[],v[]; int dis;
        scanf("%s%s%d",u,v,&dis);
        e[tot].u=get(u),e[tot].v=get(v);
        e[tot].dis=dis;
        g[get(u)].push_back(tot),tot++;

        e[tot].u=get(v),e[tot].v=get(u);
        e[tot].dis=dis;
        g[get(v)].push_back(tot),tot++;
    }

    ans1=1.0*INF;
    ans2=1.0*INF;
    MIN=;

    for(int i=n+;i<=n+m;i++)
    {
        SPFA(i);
        int sum=,MIN_NOW=INF;
        for(int j=;j<=n;j++)
        {
            if(dis[j]>ds) {sum=-;break;}
            else
            {
                sum=sum+dis[j];
                MIN_NOW=min(MIN_NOW,dis[j]);
            }
        }
        if(sum==-) continue;

        if(MIN_NOW>MIN)
        {
            MIN=MIN_NOW;
            ans2=1.0*sum/n;
            ans1=1.0*MIN;
            ans=i;
        }

        else if(MIN_NOW==MIN&&1.0*sum/n<ans2)
        {
            ans2=1.0*sum/n;
            ans=i;
        }
    }
    if(MIN==) printf("No Solution\n");
    else
    {
        printf("G%d\n",ans-n);
        printf("%.1lf %.1lf\n",ans1,ans2);
    }
    return ;
}