codeforces1301 F. Super Jaber(多源bfs+枚举)
生活随笔
收集整理的這篇文章主要介紹了
codeforces1301 F. Super Jaber(多源bfs+枚举)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
F. Super Jaber
最終答案的方案可以分為2種:
每種顏色跑多源bfs,預處理d[k][u]從顏色為k的格子出發,到達u時所需要的最小代價。
每次詢問比較上述兩種情況的答案即可。
時間復雜度O(KNM+KQ)O(KNM+KQ)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; }總結
以上是生活随笔為你收集整理的codeforces1301 F. Super Jaber(多源bfs+枚举)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 净利润增长率计算公式 净利润增长率介绍
- 下一篇: codeforces1303 F. Nu