codeforces1301 F. Super Jaber（多源bfs+枚举）

F. Super Jaber

最终答案的方案可以分为2种：

1. 不使用传送，即曼哈顿距离。
2. 从起点S到最近（步数最少）的颜色为C的格子x，传送到离终点T最近的颜色也为c的格子y，再走到T（从S到x以及y到T的过程中均使用传送）

每种颜色跑多源bfs，预处理d[k][u]从颜色为k的格子出发，到达u时所需要的最小代价。

每次询问比较上述两种情况的答案即可。

时间复杂度$O(KNM+KQ)$

#include<bits/stdc++.h>

using namespace std;
constexpr int N=1000005,K=41;
int n,m,k;
int ct;
int get(int x,int y){return (x-1)*m+y;}
int h[N],ne[4*N],e[4*N],idx;
void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;}
vector<int> clr[K];
int mp[N];
bool vis[K];
int d[K][N];
void build()
{
    int dx[]={1,-1,0,0},dy[]={0,0,1,-1};
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            for(int k=0;k<4;k++)
            {
                int a=i+dx[k],b=j+dy[k];
                if(a>=1&&b>=1&&a<=n&&b<=m) add(get(i,j),get(a,b));
            }
}
void bfs(int k)
{
    queue<int> q;
    memset(vis,0,sizeof vis);
    for(auto t:clr[k]) d[k][t]=0,q.push(t);
    vis[k]=1;
    while(q.size())
    {
        int u=q.front();q.pop();
        
        for(int i=h[u];i!=-1;i=ne[i])
        {
            int v=e[i];
            if(d[k][v]>d[k][u]+1)
            {
                d[k][v]=d[k][u]+1;
                q.push(v);
            }
        }
        
        if(vis[mp[u]]) continue;
        
        for(auto v:clr[mp[u]]) 
            if(d[k][v]>d[k][u]+1)
            {
                d[k][v]=d[k][u]+1;
                q.push(v);
            }
        vis[mp[u]]=1;
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
    int T=1;
    //cin>>T;
    while(T--)
    {
        memset(h,-1,sizeof h);idx=0;
        cin>>n>>m>>k;
        for(int i=1;i<=n;i++) 
            for(int j=1;j<=m;j++) 
            {   
                int c;
                cin>>c;
                int id=get(i,j);
                mp[id]=c;
                clr[c].push_back(id);
            }
        build();
        memset(d,0x3f,sizeof d);
        for(int i=1;i<=k;i++) bfs(i);
        
        
        cin>>ct;
        while(ct--)
        {
            int r1,c1,r2,c2;
            cin>>r1>>c1>>r2>>c2;
            int u=get(r1,c1),v=get(r2,c2);
            int ans=abs(r1-r2)+abs(c1-c2);
            for(int i=1;i<=k;i++) ans=min(ans,d[i][u]+d[i][v]+1);
            cout<<ans<<'\n';
        }
    
    }
    return 0;
}