CCF 最优配餐

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/clx55555/article/details/86820071

乍一看没有思路，看看了别人的代码，发现一个bfs救恩那个搞定。

#include<bits/stdc++.h>
#define MAX 1010
using namespace std;

int n,m,k,d;
int v[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
int vis[MAX][MAX];
long long order[MAX][MAX];
struct Node{
    int x,y;    
    long long dis;
    Node(){}
    Node(int xx,int yy,int c){
        x=xx,y=yy;
        dis=c;
    }
};
queue<Node>Q;

void init(){
    cin>>n>>m>>k>>d;
    int a,b,c;
    for(int i=0;i<m;i++){
        cin>>a>>b;
        Q.push(Node(a,b,0));
    }
    for(int i=0;i<k;i++){
        cin>>a>>b>>c;
        order[a][b]+=c;
    }
    for(int i=0;i<d;i++){
        cin>>a>>b;
        vis[a][b]=1;
    }
}

void solve(){
    long long ans=0;
    while(!Q.empty()){
        Node t=Q.front();
        Q.pop();
        int x=t.x,y=t.y,dis=t.dis;
        for(int i=0;i<4;i++){
            int xx=x+v[i][0],yy=y+v[i][1];
            if(xx>=1&&xx<=n&&yy>=1&&yy<=n&&!vis[xx][yy]){
                ans+=order[xx][yy]*(dis+1);
                vis[xx][yy]=1;
                Q.push(Node(xx,yy,dis+1));
            }
        }
    }
    cout<<ans<<endl;
}

int main(){
    init();
    solve();
    return 0;
}