五子连珠：五颗连线定输赢

fx-gomoku

local Gomoku = {}
local KEY = { N = 11 }
local DIRS = { { 1, 0 }, { 0, 1 }, { 1, 1 }, { 1, -1 } }

local function LineScore(board, c, r, me)
    local total = 0
    for _, d in ipairs(DIRS) do
        local cnt, block = 1, 0
        for s = 1, 4 do
            local cc, rr = c + d[1] * s, r + d[2] * s
            if cc < 1 or rr < 1 or cc > KEY.N or rr > KEY.N then
                block = block + 1; break
            end
            local v = board[rr][cc]
            if v == me then cnt = cnt + 1
            elseif v > 0 then block = block + 1; break end
        end
        for s = 1, 4 do
            local cc, rr = c - d[1] * s, r - d[2] * s
            if cc < 1 or rr < 1 or cc > KEY.N or rr > KEY.N then
                block = block + 1; break
            end
            local v = board[rr][cc]
            if v == me then cnt = cnt + 1
            elseif v > 0 then block = block + 1; break end
        end
        if cnt >= 5 then total = total + 100000
        elseif cnt == 4 then total = total + (block >= 2 and 800 or 8000)
        elseif cnt == 3 then total = total + (block >= 2 and 90 or 900)
        else total = total + cnt * 10 - block * 5 end
    end
    return total
end

function Gomoku.Pick(board, me, foe)
    local best, bestScore = nil, -1
    for r = 1, KEY.N do
        for c = 1, KEY.N do
            if board[r][c] == 0 then
                -- 进攻分放大一点, 先手优势
                local s = LineScore(board, c, r, me) * 1.05
                        + LineScore(board, c, r, foe)
                if s > bestScore then bestScore = s; best = { c = c, r = r } end
            end
        end
    end
    return best
end

-- 落子后由引擎回调判胜: 任一方向连满五子即胜
function Gomoku.OnPlace(c, r, me)
    for _, d in ipairs(DIRS) do
        local cnt = 1
        for s = 1, 4 do
            local cc, rr = c + d[1] * s, r + d[2] * s
            if SL:GetValue("cell", cc, rr) == me then cnt = cnt + 1 else break end
        end
        for s = 1, 4 do
            local cc, rr = c - d[1] * s, r - d[2] * s
            if SL:GetValue("cell", cc, rr) == me then cnt = cnt + 1 else break end
        end
        if cnt >= 5 then
            SL:PlayFxAt("win_line", c, r, d[1], d[2])
            SL:ShowSystemTips((me == 1 and "黑" or "白") .. "五连!")
        end
    end
end

return Gomoku