Go语言开发点餐系统中的订单分配功能实现方法

package main

import (
    "fmt"
    "sort"
)

// 骑手结构体
type Rider struct {
    ID          int    // 骑手ID
    Speed       int    // 接单速度
    Interval    int    // 接单间隔
    LocationX   int    // 骑手位置坐标X
    LocationY   int    // 骑手位置坐标Y
    AssignedNum int    // 已分配订单数量
}

// 订单结构体
type Order struct {
    ID          int    // 订单ID
    LocationX   int    // 订单位置坐标X
    LocationY   int    // 订单位置坐标Y
    DeliveryNum int    // 订单时效性
}

// 计算骑手与订单的距离
func calcDistance(rider Rider, order Order) int {
    distance := abs(rider.LocationX-order.LocationX) + abs(rider.LocationY-order.LocationY)
    return distance
}

// 绝对值函数
func abs(num int) int {
    if num < 0 {
        return -num
    }
    return num
}

// 订单分配函数
func assignOrder(riders []Rider, orders []Order) map[int][]int {
    result := make(map[int][]int)
    sort.Slice(orders, func(i, j int) bool {
        return orders[i].DeliveryNum > orders[j].DeliveryNum
    })

    for _, order := range orders {
        minDistance := 100000  // 设定一个最大距离
        assignedRiderID := -1  // 默认值为-1，表示未分配
        for _, rider := range riders {
            if rider.AssignedNum >= rider.Interval {  // 骑手接单数量超过间隔，跳过该骑手
                continue
            }
            distance := calcDistance(rider, order)
            if distance < minDistance {
                minDistance = distance
                assignedRiderID = rider.ID
            }
        }
        if assignedRiderID == -1 {  // 未找到骑手，跳过该订单
            continue
        }
        result[assignedRiderID] = append(result[assignedRiderID], order.ID)
        riders[assignedRiderID].AssignedNum++
    }

    return result
}

func main() {
    riders := []Rider{
        {ID: 1, Speed: 3, Interval: 2, LocationX: 1, LocationY: 1},
        {ID: 2, Speed: 2, Interval: 4, LocationX: 2, LocationY: 2},
        {ID: 3, Speed: 4, Interval: 3, LocationX: 3, LocationY: 3},
    }

    orders := []Order{
        {ID: 1, LocationX: 4, LocationY: 4, DeliveryNum: 5},
        {ID: 2, LocationX: 5, LocationY: 5, DeliveryNum: 2},
        {ID: 3, LocationX: 2, LocationY: 3, DeliveryNum: 4},
    }

    result := assignOrder(riders, orders)
    fmt.Println(result)
}

map[1:[2] 2:[3] 3:[1]]