【做题】计蒜客11217 百度地图的实时路况——分治

又是一道明明简单我却不会的题。

切入点当然是O(n^4)的暴力。显然这其中有大量的重复计算。

一开始的想法是从前往后，从后往前各跑一遍floyd。这样做的关键问题在于如何合并两个floyd的结果。然而我只想出了O(n^3)的合并。故这种做法除了只有暴力1/6的常数（手算得到）之外，就并没有什么卵用了。放弃这个想法。

换言之，我们的做法是需要避免多次添加同一节点。这需要我们对删去一个点的方案进行分类。

然后我就看了题解。

因为我们的结果是单个点被删去，所以考虑从从多个点被删去的状态转移过来。

记dis(l,r)的l到r中的节点尚未被添加时任意两点最短路的集合。我们就可以转移到dis(l,(l+r)/2)和dis((l+r)/2+1,r)了，并且当前状态只会被一个状态转移过来。故每个点只会被添加一次。

时间复杂度O(n^3)。

#include <bits/stdc++.h>
using namespace std;
const int N=305,INF=1e8;
int dis[10][N][N],n;
long long ans;
void floyd(int l,int r,int now)
{
    for(int k=l;k<=r;k++)for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)
        dis[now][i][j]=min(dis[now][i][k]+dis[now][k][j],dis[now][i][j]);
}
void solve(int l,int r,int now)
{
    if(l==r)
    {
        for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)if(i!=l&&j!=l)
            ans+=(dis[now][i][j]==INF?-1:dis[now][i][j]);
        return;
    }
    int mid=(l+r)>>1;
    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)
        dis[now+1][i][j]=dis[now][i][j];
    floyd(l,mid,now+1);
    solve(mid+1,r,now+1);
    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)
        dis[now+1][i][j]=dis[now][i][j];
    floyd(mid+1,r,now+1);
    solve(l,mid,now+1);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)
    {
        scanf("%d",&dis[0][i][j]);
        dis[0][i][j]=(dis[0][i][j]==-1?INF:dis[0][i][j]);
    }
    solve(1,n,0);
    printf("%lld\n",ans);
    return 0;
}

 

小结：这大概是我对floyd不够深入了解吧。