发现:只在所有自己的城市建水泵一定是最优解。
所以对自己的城市按高度排序,该城市不用建的前提是从他出发经过一条高度都小于等于他的路径能到达一个已经修建水泵的
sort+bfs......
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #include<cmath> #define N 1005 using namespace std; int dx[4]={-1,1,0,0},dy[4]={0,0,-1,1}; int bo[N][N],num,n,m,NOW,exi[N][N]; int qx[N*N],qy[N*N],g[N][N],ans; bool vis[N][N]; struct data{ int x,y,h; bool operator < (const data &a)const{return h<a.h;} }d[N*N]; void add(int x,int y,int h){d[++num].x=x;d[num].y=y;d[num].h=h;bo[x][y]=1;} bool bfs(int x,int y){ NOW++;bo[x][y]=NOW; qx[1]=x;qy[1]=y; int h=1,t=1,nx,ny,nh=g[x][y]; while(h<=t){ nx=qx[h];ny=qy[h++]; for(int i=0;i<4;i++) if(exi[nx+dx[i]][ny+dy[i]]&&bo[nx+dx[i]][ny+dy[i]]!=NOW&&g[nx+dx[i]][ny+dy[i]]<=nh){ bo[nx+dx[i]][ny+dy[i]]=NOW; qx[++t]=nx+dx[i];qy[t]=ny+dy[i]; if(vis[nx+dx[i]][ny+dy[i]])return 1; } } return 0; } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=m;j++){ exi[i][j]=1; scanf("%d",&g[i][j]); if(g[i][j]>0)add(i,j,g[i][j]); else g[i][j]=-g[i][j]; } sort(d+1,d+num+1); for(int i=1;i<=num;i++){ if(!bfs(d[i].x,d[i].y))ans++; vis[d[i].x][d[i].y]=1; } printf("%d ",ans); return 0; }