CF986A Fair

题面

解题思路

多源bfs，设dis[i][j] 设i点能获得j货物的最小长度。以每个货物跑一次bfs，更新每个点的dis值。时间复杂度理论上是nk，我貌似写成了nk(n/k)，但还是过了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-48;ch=getchar();}
    return x*f;
}

using namespace std;
const int MAXN = 100005;

int n,m,k,s,dis[MAXN][105];
int head[MAXN],cnt;
int to[MAXN<<1],nxt[MAXN<<1];
bool vis[MAXN];
vector<int> g[105];   //g[i]表示第i个货物有哪些能生产，vector大法
queue<int> q;

inline void add(int bg,int ed){
    to[++cnt]=ed,nxt[cnt]=head[bg],head[bg]=cnt;
}

inline void bfs(int kk){
    while(q.size()){
        int x=q.front();q.pop();
        for(register int i=head[x];i;i=nxt[i]){
            int u=to[i];if(vis[u]) continue;
            if(dis[u][kk]>dis[x][kk]+1){
                dis[u][kk]=dis[x][kk]+1;
                vis[u]=1;q.push(u);
            }
        }
    }
}

int main(){
    memset(dis,0x3f,sizeof(dis));
    n=rd();m=rd();k=rd();s=rd();
    int x;
    for(register int i=1;i<=n;i++) x=rd(),g[x].push_back(i);
    for(register int i=1;i<=m;i++){
        int x=rd(),y=rd();
        add(x,y);add(y,x);
    }
    for(register int i=1;i<=k;i++) {
        memset(vis,false,sizeof(vis));
        for(register int j=0;j<g[i].size();j++)
            q.push(g[i][j]),dis[g[i][j]][i]=0,vis[g[i][j]]=1;
        bfs(i);
    }
    for(register int i=1;i<=n;i++){
        sort(dis[i]+1,dis[i]+1+k);
        int ans=0;
        for(register int j=1;j<=s;j++)
            ans+=dis[i][j];
        printf("%d ",ans);
    }
    return 0;
}