201409-4 最优配餐

最新推荐文章于 2022-10-20 19:18:44 发布

清汤芥末面

最新推荐文章于 2022-10-20 19:18:44 发布

阅读量208

点赞数

分类专栏： csp 文章标签：动态规划算法 leetcode

本文链接：https://blog.csdn.net/weixin_45405831/article/details/123230879

版权

csp 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

多源最短路问题，刚开始我每个起点做BFS，会超时，增加一个超级源点即可。

#include<bits/stdc++.h>
#define x first
#define y second
#define ll long long
using namespace std;
typedef pair<int,int>PII;
const int maxn=1e3+3,INF=0x3f;
int n,m,k,s,a[maxn][maxn],d[maxn][maxn],vis[maxn][maxn];
PII store[maxn*maxn];
void bfs()
{
    int num=0;
    queue<pair<PII,int>>que;
    for(int i=0;i<m;i++)
        que.push({{store[i].x,store[i].y},0});
    while(!que.empty())
    {
        pair<PII,int> e=que.front();
        que.pop();
        PII node=e.x;
        //cout<<node.x<<" "<<node.y<<endl;
        if(node.x>n||node.x<=0||node.y<=0||node.y>n||vis[node.x][node.y]==1) continue ;
        vis[node.x][node.y]=1;
        if(a[node.x][node.y]>0)
            d[node.x][node.y]=min(d[node.x][node.y],e.y);
        if(a[node.x][node.y]==-1)
            continue;
        que.push({{node.x+1,node.y},e.y+1});
        que.push({{node.x-1,node.y},e.y+1});
        que.push({{node.x,node.y+1},e.y+1});
        que.push({{node.x,node.y-1},e.y+1});
    }
}
int main()
{
    cin>>n>>m>>k>>s;
    memset(d,INF,sizeof(d));
    for(int i=0;i<m;i++)
        {
            cin>>store[i].x>>store[i].y;
            d[store[i].x][store[i].y]=0;
        }
    for(int i=0;i<k;i++)
        {
            int xx,yy,c;
            cin>>xx>>yy>>c;
            a[xx][yy]+=c;
        }
    for(int i=0;i<s;i++)
        {
            int xx,yy;
            cin>>xx>>yy;
            a[xx][yy]=-1;
        }
    bfs();
    ll sum=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        if(a[i][j]>0)
            sum+=(ll)d[i][j]*(ll)a[i][j];
    cout<<sum<<endl;
    return 0;
}