CodeForces - 986A Fair (BFS+贪心)

Posted xiuwenli

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了CodeForces - 986A Fair (BFS+贪心)相关的知识,希望对你有一定的参考价值。

题意:有N个点M条边的无向图,每个点有给定的ai(1<=ai<=K,K<=200)表示该点拥有的物品编号,保证1-K在N个点全部出现。求每个点收集S个不同的物品所要走过的最短路程(边的长度为1)。

分析:N是1e5,如果直接对每个点搜索肯定超时。发现K的范围很小,而且1-K全部覆盖。那么考虑对所有1-K的值BFS,用一个二维数组dp[i][j]记录i点要获取编号为j的物品最少走过的路程,并对每个点取最小的S个物品对应的路径。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn =1e5+5;
int a[maxn];
int res[maxn];
vector<int> G[maxn];
bool vis[maxn];
int d[maxn][305];
void init(int N)    {for(int i=1;i<=N;++i) G[i].clear();}
void AddEdge(int u,int v){G[u].push_back(v);}

void BFS(int val,int N)
{
    memset(vis,0,sizeof(vis));
    queue<int> Q;
    for(int i=1;i<=N;++i){
        if(a[i]==val){
            vis[i]=true;
            Q.push(i);
        }
    }
    while(!Q.empty()){
        int x =Q.front();Q.pop();
        for(int i=0;i<G[x].size();++i){
            int v = G[x][i];
            if(!vis[v]){
                vis[v] = true;
                d[v][val] = d[x][val]+1;
                Q.push(v);
            }
        }
    }
}

int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int N,M,K,S,u,v;
    while(scanf("%d%d%d%d",&N,&M,&K,&S)==4){
        init(N);
        memset(d,0,sizeof(d));
        for(int i=1;i<=N;++i) scanf("%d",&a[i]);
        for(int i=1;i<=M;++i){
            scanf("%d%d",&u,&v);
            AddEdge(u,v);
            AddEdge(v,u);
        }
        for(int i=1;i<=K;++i){
            memset(vis,0,sizeof(vis));
            BFS(i,N);
        }
        for(int i=1;i<=N;++i){
            sort(d[i]+1,d[i]+K+1);
            res[i]=0;
            for(int j =1;j<=S;++j){
                res[i]+=d[i][j];
            }
        }
        for(int i=1;i<N;++i) printf("%d ",res[i]);
        printf("%d
",res[N]);
    }
    return 0;
}

 

以上是关于CodeForces - 986A Fair (BFS+贪心)的主要内容,如果未能解决你的问题,请参考以下文章

CF 986A Fair——多源bfs

CF 986A Fair(多源BFS)

CodeForces - 864A Fair Game

Codeforces Round #526 C - The Fair Nut and String /// 组合递推

Codeforces Round #526 (Div. 2) C. The Fair Nut and String

Codeforces Round #485 (Div. 2) D. Fair