建物流中转站
题目描述
Shopee物流会有很多个中转站。在选址的过程中,会选择离用户最近的地方建一个物流中转站。
假设给你一个二维平面网格,每个格子是房子则为1,或者是空地则为0。找到一个空地修建一个物流中转站,使得这个物流中转站到所有的房子的距离之和最小。 能修建,则返回最小的距离和。如果无法修建,则返回 -1。
若范围限制在100*100以内的网格,如何计算出最小的距离和?
当平面网格非常大的情况下,如何避免不必要的计算?
解题思路:
本题数据并不大,暴力枚举完全可以解决问题,但是这样写显然没什么技术含量。当数据过大,本题一般方法会超时,我们可以建立二维前缀和来计算距离。
二维前缀:
每一点的含义可以表述为:设Dp[i][j]为右下顶点,以(1,1)为左上顶点所成的矩形中有多少个中转站。很容易得状态方程:Dp[i][j] = Dp[i][j - 1] + Dp[i - 1][j] - Dp[i - 1][j - 1] + a[i][j];(在草稿纸上画一下就可以直观地看出来)
计算方式:
暴力枚举出顶点(1,1)到其他仓库的距离和,每一次移动有两种移动方式:
横向向右移动和竖直向下移动。状态方程分别为:
sum2 += 2 * Dp[n][j - 1] - Dp[n][n];
sum1 += 2 * Dp[i - 1][n] - Dp[n][n];
sum 表示该点到其他点的距离和。
代码示例:
#include<bits/stdc++.h>
using namespace std;
int a[101][101] = { 0 }, Dp[101][101] = { 0 };
int main()
{
int n,flag=0;
cin >> n;
for(int i=1;i<=n;i++)
for (int j = 1; j <= n; j++)
{
int k;
cin >> k;
if (k)
{
a[i][j] = 1;
}
}
int sum = 0;
for (int i = 1; i <= n; i++)//使用状态方程时需初始化第一行和第一列的值
{
if (a[1][i])
{
sum++;
}
Dp[1][i] = sum;
}
sum = 0;
for (int i = 1; i <= n; i++)
{
if (a[i][1])
{
sum++;
}
Dp[i][1] = sum;
}
for (int i = 2; i <= n; i++)//计算Dp[i][j]
for (int j = 2; j <= n; j++)
Dp[i][j] = Dp[i][j - 1] + Dp[i - 1][j] - Dp[i - 1][j - 1] + a[i][j];
sum = 0;
for (int i = 1; i <= n; i++)//计算出(1,1)到其他点的距离
for (int j = 1; j <= n; j++)
{
if(a[i][j])
sum += abs(i - 1) + abs(j - 1);
}
int sum1,sum2;
sum1 = sum2 = sum;
for (int i = 1; i <= n; i++)
{//这里需要注意的是当每一行计算完毕后需要以该行第一个位置的值来计算下一行第一个位置的值!
if (i != 1)
{
sum1 += 2 * Dp[i - 1][n] - Dp[n][n];
}
if (!a[i][1])
sum = min(sum1, sum),flag=1;
sum2 = sum1;
for (int j = 2; j <= n; j++)
{
sum2 += 2 * Dp[n][j - 1] - Dp[n][n];
if (!a[i][j])
sum = min(sum2, sum),flag=1;
}
}
if (flag)
cout << sum;
else
cout << "-1";
return 0;
}