package?AprioriTest;
import?java.io.BufferedReader;
import?java.io.File;
import?java.io.FileNotFoundException;
import?java.io.FileReader;
import?java.io.IOException;
import?java.util.ArrayList;
import?java.util.HashMap;
import?java.util.List;
import?java.util.Map;
import?java.util.Set;
import?java.util.TreeSet;
/**
?*?功能：Apriori算法的實現
?*/
public?class?AprioriTest01?{
?
?private?int?minSup;
?private?static?List?data;
?private?static?List>?dataSet;
?
?public?static?void?main（String[]?args）?{
???
??long?startTime?=?System.currentTimeMillis（）;
??AprioriTest01?apriori?=?new?AprioriTest01（）;
??File?file?=?new?File（“G:/mf_2015511153514740.txt“）;
???//設置最小支持度
??apriori.setMinSup（150）;
??//構造數據集
??data?=?apriori.buildData（file）;
??//構造頻繁1項集
??List>?f1Set?=?apriori.findF1Item（data）;
?//?apriori.printSet（f1Set?1）;
??List>?result?=?f1Set;?
??int?i?=?2;
??do{
???result?=?apriori.arioriGen（result）;
???apriori.printSet（result?i）;
???i++;
??}while（result.size（）?!=?0）;
??long?endTime?=?System.currentTimeMillis（）;
??System.out.println（“共用時：?“?+?（endTime?-?startTime）?+?“ms“）;
?}
?public?void?setMinSup（int?minSup）{
??this.minSup?=?minSup;
?}
?
?/**
??*?構造原始數據集，
??*?使用文件中的數據集
??*/
?List?buildData（File?fileName）{
??List?data?=?new?ArrayList（）;
???try{
????BufferedReader?reader?=?new?BufferedReader（new?FileReader（fileName））;
????String?line;
????while（?（?line?=?reader.readLine（））?!=?null?）{
?????data.add（line）;
????}
???}catch?（FileNotFoundException?e）{
????e.printStackTrace（）;
???}catch?（IOException?e）{
????e.printStackTrace（）;
???}
??
??dataSet?=?new?ArrayList>（）;
??Set?dSet;
??for（String?d?:?data）{
???dSet?=?new?TreeSet（）;
???String[]?dArr?=?d.split（“?“）;
???for（String?str?:?dArr）{
????dSet.add（str）;
???}
???dataSet.add（dSet）;
??}
??return?data;
?}
?
?/**
??*?找出候選1項集
??*?@param?data
??*?@return?result
??*/
?List>?findF1Item（List?data）{
??List>?result?=?new?ArrayList>（）;
??Map?dc?=?new?HashMap（）;
??for（String?d?:?data）{
???String[]?items?=?d.split（“?“）;
???for（String?item?:?items）{
????if（dc.containsKey（item））?{
?????dc.put（item?dc.get（item）+1）;
????}else{
?????dc.put（item?1）;
????}
???}
??}
??Set?itemKeys?=?dc.keySet（）;
??Set?tempKeys?=?new?TreeSet（）;
??for（String?str?:?itemKeys）{
???tempKeys.add（str）;
??}
??
??for（String?item?:?tempKeys）{
???if（dc.get（item）?>=?minSup）?{
????Set?f1Set?=?new?TreeSet（）;
????f1Set.add（item）;
????result.add（f1Set）;
???}
??}?
??return?result;
?}
?
?/**
??*?利用arioriGen方法由k-1項集生成k項集
??*@param?preSet
??*@return
??*
??*/
?List>?arioriGen（List>?preSet）?{
??
??List>?result?=?new?ArrayList>（）;
??int?preSetSize?=?preSet.size（）;
??
??for（int?i?=?0;?i?