package timeflow.util;
|
|
import java.util.*;
|
|
public class DoubleBag<T> implements Iterable<T> {
|
private HashMap<T, Count> table;
|
private double max;
|
|
public DoubleBag()
|
{
|
table=new HashMap<T, Count>();
|
}
|
|
public double getMax()
|
{
|
return max;
|
}
|
|
public List<T> listTop(int n, boolean useSum)
|
{
|
int count=0;
|
Iterator<T> i=list(useSum).iterator();
|
List<T> top=new ArrayList<T>();
|
while (count<n && i.hasNext())
|
{
|
top.add(i.next());
|
count++;
|
}
|
return top;
|
}
|
|
public List<T> list(final boolean useSum)
|
{
|
List<T> result=new ArrayList<T>();
|
result.addAll(table.keySet());
|
|
Collections.sort(result, new Comparator<T>()
|
{
|
public int compare(T x, T y)
|
{
|
double d= useSum ? num(y)-num(x) : average(y)-average(x);
|
return d>0 ? 1 : (d<0 ? -1 : 0);
|
}
|
});
|
return result;
|
}
|
|
public double num(T x)
|
{
|
Count c=table.get(x);
|
if (c!=null)
|
return c.num;
|
else
|
return 0;
|
}
|
|
public double average(T x)
|
{
|
Count c=table.get(x);
|
return c.num/c.vals;
|
}
|
|
public void add(T x, double z)
|
{
|
if (Double.isNaN(z))
|
return;
|
Count c=table.get(x);
|
double sum=z;
|
if (c!=null)
|
{
|
c.add(z);
|
sum=c.num;
|
}
|
else
|
{
|
table.put(x, new Count(z));
|
}
|
max=Math.max(sum, max);
|
}
|
|
class Count
|
{
|
double num;
|
int vals;
|
|
public Count(double num)
|
{
|
this.num=num;
|
vals=1;
|
}
|
|
public double add(double x)
|
{
|
vals++;
|
return num+=x;
|
}
|
}
|
|
public int size()
|
{
|
return table.size();
|
}
|
|
|
public List<T> unordered()
|
{
|
List<T> result=new ArrayList<T>();
|
result.addAll(table.keySet());
|
return result;
|
}
|
|
|
public int removeLessThan(int cut)
|
{
|
|
Set<T> small=new HashSet<T>();
|
for (T x: table.keySet())
|
{
|
if (num(x)<cut)
|
small.add(x);
|
}
|
for (T x:small)
|
table.remove(x);
|
return small.size();
|
}
|
|
public static void main(String[] args)
|
{
|
DoubleBag<String> b=new DoubleBag<String>();
|
b.add("a",1);
|
b.add("b",2);
|
b.add("a",3);
|
System.out.println(b.num("a"));
|
System.out.println(b.num("b"));
|
System.out.println(b.num("c"));
|
List<String> s=b.list(true);
|
for (int i=0; i<s.size(); i++)
|
System.out.println(s.get(i)+": "+b.num(s.get(i)));
|
}
|
|
@Override
|
public Iterator<T> iterator() {
|
return table.keySet().iterator();
|
}
|
}
|
