import java.lang.Math; import java.util.ArrayList; import java.util.Random; public class Program { // centroids are indicated with upper case letter while data point are indicated with lower case letter (the centroid it's attached in lowercase) public static void main(String[] args) { ML m = new ML(); m.start(20,10,10);// time limit xceed each time } } class ML { ArrayList data= new ArrayList(); ArrayList centr= new ArrayList(); int large=20;// must be at least the same of the forst argument: is how long the "draw" will be public ML(){ } public void init(int n,int c)//initialize: generate random data and random centroids { for (int i=0;ir) { min=r; c=centr.get(j); } } Pos t = data.get(i); t.c=c; data.set(i,t); } } public void moveCentr()//move centroids in th average position of his data point { for (int j=0;j Rdata=data; ArrayList Rcentr=centr; for (int i=0;i> sD = sort(Rdata); ArrayList> sC = sort(Rcentr); System.out.println("D"); for (int line=0;line Rdata=data; //rounded data (in int) ArrayList Rcentr=centr;// rounded centr for (int i=0;i> sort(ArrayList l)// sort per Y and per X { ArrayList> res =new ArrayList>(); for (int line=0;line()); ArrayList tLine=res.get(line);//actual line (Pos) for (int i=0;itLine.get(j+1).x)// bubble sort per sorting x { Pos t = tLine.get(j); tLine.set(j,tLine.get(j+1)); tLine.set(j+1,t); } } } res.set(line,tLine); } return res; } public Pos[][] toArray(ArrayList> a)// change from list matrix to normal matrix { Pos[][] res = new Pos[large][large]; for (ArrayList j:a) { for (Pos i:j)// loop each item { res[(int)i.y][(int)i.x]=i;//put it into the matrix } } return res; } } class Pos //class pos each datapoint and centroids are poses { public String name;//name (for centroids only) public float x,y; public Pos c;//centroids (for data point only) public Pos(float x,float y)//normal data point contructor { this.x=x; this.y=y; } public Pos(float x,float y,String name)//normal centroids contructor { this.x=x; this.y=y; this.name=name; } public float dis(Pos p)//calculate distance { float dx=Math.abs(p.x-x); float dy=Math.abs(p.y-y); return (float)Math.sqrt(dx*dx+dy*dy); } public boolean equals(Pos p)// equals, return boolean { if (x==p.x&&y==p.y) { return true; } return false; } }

K means + visual