码迷,mamicode.com
首页 > 其他好文 > 详细

Fair CodeForces - 987D(巧妙bfs)

时间:2018-08-23 16:54:32      阅读:138      评论:0      收藏:0      [点我收藏+]

标签:size   商品   生产   std   ems   void   pop   排序   its   

题意:

  有n个城市 m条边,每条边的权值为1,每个城市生产一种商品(可以相同,一共k种),求出分别从每个城市出发获得s种商品时所走过路的最小权值

解析:

  我们倒过来想,不用城市找商品,而是商品找城市,求出每个商品到达每个城市的最短路,w[i][j]即为生产商品j的城市到达城市i的最短路,最后对于每个i排序w[i] 取前s个小的即可

 

#include <bits/stdc++.h>
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;
const int maxn = 1001000, INF = 0x7fffffff;
typedef long long LL;
vector<int> G[maxn];
int n, m, k, s;
int w[maxn][105];   //w[i][j] 生产j商品的城市到i城市的最短路

void bfs(int x)
{
    queue<int> Q;
    Q.push(x+n);
    while(!Q.empty())
    {
        int u = Q.front(); Q.pop();
        for(int i=0; i<G[u].size(); i++)
        {
            int v = G[u][i];
            if(w[v][x] == 0)
            {
                w[v][x] = w[u][x] + 1;      //目的就是找到生产j商品的城市(无论哪一个)到i城市的最短路
                Q.push(v);
            }
        }
    }
}

int main()
{
    int u, v, c;
    scanf("%d%d%d%d", &n, &m, &k, &s);
    for(int i=1; i<=n; i++)
    {
        scanf("%d", &c);
        G[c+n].push_back(i);
    }
    for(int i=0; i<m; i++)
    {
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for(int i=1; i<=k; i++) bfs(i);
    for(int i=1; i<=n; i++)
    {
        int res = 0;
        sort(w[i]+1, w[i]+k+1);     //把k个商品分别到i城市的最短路排序,累加最小的s个
        for(int j=1; j<=s; j++)
            res += w[i][j] - 1;     //因为每次是从商品开始广搜的 每个商品多加了一个1
        printf("%d ", res);
    }


    return 0;
}

 

Fair CodeForces - 987D(巧妙bfs)

标签:size   商品   生产   std   ems   void   pop   排序   its   

原文地址:https://www.cnblogs.com/WTSRUVF/p/9524234.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!