用户分群

package main

import (
	"fmt"
	"math"
)

func main() {
	k := 0

	for {
		n, err := fmt.Scan(&k)
		if n == 0 {
			break
		}
		if err != nil {
			return
		}
		if k == 0 {
			break
		}
		specialList := make([][3]float64, k)
		for i := 0; i < k; i++ {
			fmt.Scan(&specialList[i][0], &specialList[i][1], &specialList[i][2])
		}
		iteCnt := 0
		fmt.Scan(&iteCnt)

		dataCnt := 0
		fmt.Scan(&dataCnt)

		dataList := make([][3]float64, dataCnt)
		for i := 0; i < dataCnt; i++ {
			fmt.Scan(&dataList[i][0], &dataList[i][1], &dataList[i][2])
		}

		for i := 0; i < iteCnt; i++ {
			groupMap := map[int][][3]float64{}

			for j := 0; j < dataCnt; j++ {
				// 计算与特征值距离 划分组别
				minVal := math.MaxFloat64
				minIdx := 0
				for idx, special := range specialList {
					val := (dataList[j][0]-special[0])*(dataList[j][0]-special[0]) +
						(dataList[j][1]-special[1])*(dataList[j][1]-special[1]) +
						(dataList[j][1]-special[1])*(dataList[j][1]-special[1])
					if val < minVal {
						minIdx = idx
						minVal = val
					}
				}
				groupMap[minIdx] = append(groupMap[minIdx], dataList[j])

			}
			// 更新特征值
			for idx, list := range groupMap {
				sum := [3]float64{}
				for _, item := range list {
					sum[0] += item[0]
					sum[1] += item[1]
					sum[2] += item[2]
				}
				specialList[idx] = [3]float64{sum[0] / float64(len(list)), sum[1] / float64(len(list)), float64(sum[2] / float64(len(list)))}
			}
		}
		for _, special := range specialList {
			fmt.Printf("%.2f %.2f %.2f\n", special[0], special[1], special[2])
		}
	}
}