poj 3316 Snakes on a Plane

題意：先說蛇的定義，蛇是由聯通的1組成，並且除了兩端，每個1只有兩個相鄰的位置是1。最大蛇的定義，不能給一條蛇在端點添加一個1使蛇更長，而不違背蛇的定義且不與其他的蛇相連接。問最大蛇的數量。

整體思路就是先用dfs遍歷聯通塊，並判斷這個聯通塊是不是蛇，然後對於每條蛇再判斷能不能再添加一個1，感覺判斷能不能添加的地方寫的有點複雜了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n, m;
int vis[205][205];
char mp[205][205];
struct snake{
    int r[3], c[3];
    int nx, isn;
}sn[20005];
int dr[] = {0, 0, 1, -1};
int dc[] = {1, -1, 0, 0};
void dfs(int r, int c, int u) {
    vis[r][c] = u;
    int t = 0;
    for(int i = 0; i < 4; i++) {
        int nr = r + dr[i];
        int nc = c + dc[i];
        if(nr >= 0 && nr < n && nc >= 0 && nc < m && mp[nr][nc] == '1') {
            t++;
            if(vis[nr][nc] == -1)
                dfs(nr, nc, u);
        }
    }
    if(t == 0) {
        sn[u].r[0] = r;
        sn[u].c[0] = c;
        sn[u].nx++;
        sn[u].isn = 2;
    }
    if(t == 1) {
        sn[u].r[sn[u].nx] = r;
        sn[u].c[sn[u].nx] = c;
        sn[u].nx++;
        sn[u].isn++;
    }
    if(t > 2)
        sn[u].isn = 3;
}
int main() {
    int i, j;
    while(~scanf("%d%d", &n, &m) && (n || m)) {
        memset(vis, -1, sizeof(vis));
        memset(sn, 0, sizeof(sn));
        for(i = 0; i < n; i++) {
            scanf("%s", mp[i]);
        }
        int cnt = 0;
        for(i = 0; i < n; i++) {
            for(j = 0; j < m; j++) {
                if(mp[i][j] == '1' && vis[i][j] < 0) {
                    dfs(i, j, cnt++);
                }
            }
        }
        int ans = 0;
        for(i = 0; i < cnt; i++) {
            if(sn[i].isn != 2) continue;
            for(j = 0; j < sn[i].nx; j++) {
                int t = 0;
                for(int k = 0; k < 4; k++) {
                    int r = sn[i].r[j] + dr[k];
                    int c = sn[i].c[j] + dc[k];
                    if(r < 0 || r >= n || c < 0 || c >= m || mp[r][c] == '1') {
                        t++;
                        continue;
                    }
                    int sum = 0;
                    for(int l = 0; l < 4; l++) {
                        int nr = r + dr[l];
                        int nc = c + dc[l];
                        if(nr == sn[i].r[j] && nc == sn[i].c[j]) continue;
                        if(nr >= 0 && nr < n && nc >= 0 && nc < m && mp[nr][nc] == '1' )
                            sum = 1;
                    }
                    if(sum)
                        t++;
                }
                if(t < 4) break;
            }
            if(j == sn[i].nx) {
                ans++;
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}