โจทย์
K-means หากลุ่มในข้อมูลที่ไม่มีป้ายกำกับ เริ่มจากเซนทรอยด์ที่กำหนด แล้วทำซ้ำ iterations ครั้ง:
- กำหนดกลุ่ม: ใส่แต่ละจุดในกลุ่มของเซนทรอยด์ที่ใกล้ที่สุด (ระยะทางแบบยุคลิด) ถ้าเซนทรอยด์สองตัวใกล้เท่ากัน ให้เลือกตัวที่มีดัชนีน้อยกว่า
- ปรับปรุง: ย้ายเซนทรอยด์แต่ละตัวไปที่ค่าเฉลี่ยของจุดที่อยู่ในกลุ่มของมัน เซนทรอยด์ที่ไม่มีจุดเลยให้อยู่ที่เดิม
จงเขียน k_means(points, centroids, iterations)
อินพุต
points: จุดข้อมูล แต่ละจุดเป็นลิสต์ของพิกัด
centroids: เซนทรอยด์เริ่มต้น k ตัว ที่มีจำนวนพิกัดเท่ากับจุด
iterations: จำนวนรอบของการกำหนดกลุ่มและปรับปรุง
เอาต์พุต
คืนค่าเซนทรอยด์สุดท้ายตามลำดับเดิม
เงื่อนไข
1 ≤ จำนวนจุด ≤ 1,000, 1 ≤ k ≤ 20, 1 ≤ จำนวนมิติ ≤ 10
0 ≤ iterations ≤ 100
-1,000 ≤ แต่ละพิกัด ≤ 1,000
ตัวอย่าง
อินพุต:
points = [
[1.0, 1.0],
[1.5, 2.0],
[8.0, 8.0],
[9.0, 8.5]
]
centroids = [
[0.0, 0.0],
[10.0, 10.0]
]
iterations = 1
ผลลัพธ์ที่คาดหวัง:
expected = [
[1.25, 1.5],
[8.5, 8.25]
]
อินพุต:
std::vector<std::vector<double>> points = {
{1.0, 1.0},
{1.5, 2.0},
{8.0, 8.0},
{9.0, 8.5}
};
std::vector<std::vector<double>> centroids = {
{0.0, 0.0},
{10.0, 10.0}
};
long long iterations = 1;
ผลลัพธ์ที่คาดหวัง:
std::vector<std::vector<double>> expected = {
{1.25, 1.5},
{8.5, 8.25}
};
อินพุต:
double[][] points = new double[][] {
new double[] {1.0, 1.0},
new double[] {1.5, 2.0},
new double[] {8.0, 8.0},
new double[] {9.0, 8.5}
};
double[][] centroids = new double[][] {
new double[] {0.0, 0.0},
new double[] {10.0, 10.0}
};
long iterations = 1;
ผลลัพธ์ที่คาดหวัง:
double[][] expected = new double[][] {
new double[] {1.25, 1.5},
new double[] {8.5, 8.25}
};
คำอธิบาย
สองจุดแรกอยู่ใกล้ (0, 0) มากกว่า และสองจุดสุดท้ายอยู่ใกล้ (10, 10) มากกว่า เซนทรอยด์ใหม่คือค่าเฉลี่ย: ((1 + 1.5) / 2, (1 + 2) / 2) = (1.25, 1.5) และ ((8 + 9) / 2, (8 + 8.5) / 2) = (8.5, 8.25)
อินพุต:
points = [
[0.0],
[1.0],
[10.0]
]
centroids = [
[0.0],
[100.0]
]
iterations = 2
ผลลัพธ์ที่คาดหวัง:
expected = [
[3.666667],
[100.0]
]
อินพุต:
std::vector<std::vector<double>> points = {
{0.0},
{1.0},
{10.0}
};
std::vector<std::vector<double>> centroids = {
{0.0},
{100.0}
};
long long iterations = 2;
ผลลัพธ์ที่คาดหวัง:
std::vector<std::vector<double>> expected = {
{3.666667},
{100.0}
};
อินพุต:
double[][] points = new double[][] {
new double[] {0.0},
new double[] {1.0},
new double[] {10.0}
};
double[][] centroids = new double[][] {
new double[] {0.0},
new double[] {100.0}
};
long iterations = 2;
ผลลัพธ์ที่คาดหวัง:
double[][] expected = new double[][] {
new double[] {3.666667},
new double[] {100.0}
};
คำอธิบาย
รอบที่ 1: (100) อยู่ไกลจากทุกจุด ทั้งสามจุดจึงเข้ากลุ่มของเซนทรอยด์ 1 ซึ่งย้ายไปที่ (0 + 1 + 10) / 3 ≈ 3.666667 ส่วนเซนทรอยด์ 2 ไม่มีจุดเลยจึงอยู่ที่ 100 เหมือนเดิม รอบที่ 2 กำหนดกลุ่มแบบเดียวกัน ผลลัพธ์จึงเป็น [[3.666667], [100]]
แหล่งอ้างอิง
โจทย์นี้ปรับให้เหมาะกับ CodeUtility กรุณาทำตามคำชี้แจงและข้อจำกัดด้านบน