最优配餐

[CCF]最优配餐

Question

　栋栋最近开了一家餐饮连锁店，提供外卖服务。随着连锁店越来越多，怎么合理的给客户送餐成为了一个急需解决的问题。
　　栋栋的连锁店所在的区域可以看成是一个n×n的方格图（如下图所示），方格的格点上的位置上可能包含栋栋的分店（绿色标注）或者客户（蓝色标注），有一些格点是不能经过的（红色标注）。
　　方格图中的线表示可以行走的道路，相邻两个格点的距离为1。栋栋要送餐必须走可以行走的道路，而且不能经过红色标注的点。

　　送餐的主要成本体现在路上所花的时间，每一份餐每走一个单位的距离需要花费1块钱。每个客户的需求都可以由栋栋的任意分店配送，每个分店没有配送总量的限制。
　　现在你得到了栋栋的客户的需求，请问在最优的送餐方式下，送这些餐需要花费多大的成本。

　输入的第一行包含四个整数n, m, k, d，分别表示方格图的大小、栋栋的分店数量、客户的数量，以及不能经过的点的数量。
　　接下来m行，每行两个整数xi, yi，表示栋栋的一个分店在方格图中的横坐标和纵坐标。
　　接下来k行，每行三个整数xi, yi, ci，分别表示每个客户在方格图中的横坐标、纵坐标和订餐的量。（注意，可能有多个客户在方格图中的同一个位置）
　　接下来d行，每行两个整数，分别表示每个不能经过的点的横坐标和纵坐标。

输出一个整数，表示最优送餐方式下所需要花费的成本。

Input

10 2 3 3
1 1
8 8
1 5 1
2 3 3
6 7 2
1 2
2 2
6 8

Output

29

Solution

// http://118.190.20.162/view.page?gpid=T13
#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<set>
#include<cctype>
#include<map>
#include<utility>
using namespace std;
#define print(A) cout << #A << ": "<< A << endl;
#define inf 0x3fffffff
typedef long long int ll;


struct node {
  ll x, y, count, dist;
  node(ll x_ = -1, ll y_ = -1, ll c = 0, ll d = -1): x(x_), y(y_), count(c), dist(d) {}
};
queue<node> q;
vector<vector<node> > mp;
vector<vector<bool> > isvisited;
ll n, m, k, d;
ll res = 0;
int direction[4][2] = {
  {1, 0},
  {-1, 0},
  {0, 1},
  {0, -1}
};
bool inMap(int x, int y) { return x <= n && x >= 1 && y <= n && y >= 1; }
void bfs() {
  while (!q.empty()) {
    node top = q.front(); q.pop();
    for (int i = 0; i < 4; i++) {
      ll x = top.x+direction[i][0];
      ll y = top.y+direction[i][1];
      if (inMap(x, y) && !isvisited[x][y]) {
        mp[x][y].dist = top.dist+1;
        // 发现自己写了很久的bfs都写错了，应该在这里设置一个已访问而不是在上面
        isvisited[x][y] = true;
        q.push(mp[x][y]);
        res += mp[x][y].dist*mp[x][y].count;
      }
    }
  }
}

int main(int argc, char const *argv[]) {
  scanf("%lld %lld %lld %lld", &n, &m, &k, &d);
  mp.resize(n+1, vector<node>());
  for (int i = 0; i < n+1; i++) mp[i].resize(n+1);

  for (int i = 1; i <= n; i++) {
    for (int j = 1; j <= n; j++) {
      mp[i][j] = node(i, j, 0, -1);
    }
  }
  isvisited.resize(n+1, vector<bool>());
  for (int i = 0; i < n+1; i++) isvisited[i].resize(n+1, false);

  for (int i = 0; i < m; i++) {
    ll x, y; scanf("%lld %lld", &x, &y);
    mp[x][y].dist = 0;
    q.push(mp[x][y]);
    isvisited[x][y] = true;
  }
  for (int i = 0; i < k; i++) {
    ll x, y, count; scanf("%lld %lld %lld", &x, &y, &count);
    mp[x][y].count += count;
  }
  // 不能走的地方统一让count变成-1
  for (int i = 0; i < d; i++) {
    ll x, y; scanf("%lld %lld", &x, &y);
    // 这些是不能访问的结点，那就直接标记已经访问过
    isvisited[x][y] = true;
  }

  bfs();
  printf("%lld\n", res);
  return 0;
}

思路：这题是特殊的BFS，原因是将多个配送点认为是起点。因此这里的初始值有多个而不只一个，其他一样。还有一个技巧，就是将直接不能访问的点可以直接设置为已访问过就好，后面就不需要进行判断了。