PAT1012 数字分类

package main

import (
    "fmt"
)

func main() {
    var n int
    fmt.Scanf("%d", &n)
    
    nums := make([]int, n)
    for i := 0; i < n; i++ {
        fmt.Scanf("%d", &nums[i])
    }

    a1 := 0
    a2, sign, flag := 0, 1, false 
    a3 := 0
    a4 := make([]int, 0)
    a5 := -1

    for _, num := range nums {
        m := num % 5
        if m == 0 {
            if num %2 == 0 {
                a1 += num
            }
        } else if m == 1 {
            flag = true
            a2 = a2 + num*sign
            sign = -sign
        } else if m == 2 {
            a3++
        } else if m == 3 {
            a4 = append(a4, num)
        } else if m == 4 {
            a5 = max(a5, num)
        }
    }

    if a1 == 0 {
        fmt.Print("N")
    } else {
        fmt.Print(a1)
    }
    
    if flag == false {
        fmt.Print(" N")
    } else {
        fmt.Printf(" %d", a2)
    }
    
    if a3 == 0 {
        fmt.Print(" N")
    } else {
        fmt.Printf(" %d", a3)
    }

    if len(a4) == 0 {
        fmt.Print(" N")
    } else {
        sum := 0
        for _, num := range a4 {
            sum += num
        }
        fmt.Printf(" %.1f", float64(sum)/float64(len(a4)))
    }

    if a5 == -1 {
        fmt.Print(" N")
    } else {
        fmt.Printf(" %d", a5)
    }
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}