洛谷P3057 [USACO12NOV]远处的牧场Distant Pastures

洛谷P3057 [USACO12NOV]远处的牧场Distant Pastures
因为是稀疏图（一个点最多连接4条边)
所以每个点跑SPFA就行了

#include <bits/stdc++.h> 
#define LL long long 
#define For(i,j,k) for(int i=j;i<=k;i++) 
#define Dow(i,j,k) for(int i=j;i>=k;i--) 
using namespace std ; 

const int N = 911 ; 
const LL inf = 1e15 ; 
const int dx[]={0,1,0,0,-1} ; 
const int dy[]={0,0,1,-1,0} ; 
int n,cnt,h,t,A,B ; 
char s[33][33] ; 
struct node{
    int to,pre,val ; 
}e[N*4];
int q[200011],head[N] ; 
LL dist[N],Mx ; 
bool visit[N] ; 

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline void add(int x,int y,int val)
{
    e[++cnt].to = y ; 
    e[cnt].val = val ; 
    e[cnt].pre = head[x] ; 
    head[x] = cnt ; 
}

inline void SPFA(int s) 
{
    For(i,0,n) dist[i] = inf ; 
    For(i,0,n) visit[i] = 0 ; 
    h = 0 ; t = 0 ; 
    q[++t] = s ; dist[s] = 0 ; 
    while(h<t) {
        int u=q[++h] ; 
        visit[u] = 0 ; 
         for(int i=head[u];i;i=e[i].pre) {
            int v = e[i].to ; 
            if(dist[u]+e[i].val<dist[v]) {
                dist[v]=dist[u]+e[i].val ; 
                if(!visit[v]) {
                    visit[v] = 1 ; 
                    q[++t] = v ; 
                }
            }
        }
    }
    For(i,1,n) 
        if(dist[i]>Mx) Mx = dist[i] ; 
}

int main() 
{
     n = read() ; A = read() ; B = read() ; 
     For(i,1,n) scanf("%s",s[i]+1) ; 
    For(i,1,n) 
      For(j,1,n)
        For(k,1,4) {
            int xx=i+dx[k] ,yy=j+dy[k],val ; 
            if(xx<1||xx>n||yy<1||yy>n) continue ; 
            if(s[i][j]==s[xx][yy]) val = A ; 
            else val=B ; 
            add((i-1)*n+j,(xx-1)*n+yy,val) ;  
         }
    Mx = -1 ;
    n*=n ;  
    For(i,1,n) SPFA( i ) ; 
    printf("%d
",Mx) ; 
    return 0 ; 
}