#pragma once

#include "stdafx.h"

static const char*	DB_NAME = "Apriori";
static const char*	HOST = "localhost";
static const char*	USER = "root";
static const char*	PASSWD = "scorpio";
static const int	MaxSize = 1000;
static const	string::size_type npos = -1;


class CAprioriSet
{
//數(shù)據(jù)訪問
private:
	Connection	con;
	Result		res;

//變量
public:	
	double	freq;						// 最小頻繁數(shù)
	int	itemCount;					// 設(shè)置數(shù)據(jù)庫中總共有的各種不同項目個數(shù)	

	int	candFreqCount[MaxSize][MaxSize];		// 各級候選項目的頻繁數(shù)	
	int	freqCount[MaxSize][MaxSize];			// 各級項目的頻繁數(shù)
	string	candlargeItem[MaxSize][MaxSize];		// 存放各級候選頻繁項目
	string	largeItem[MaxSize][MaxSize];			// 存放各級頻繁項目

	//vector<bool>	candLargeItemFlag;
	vector<int>	largeItemCount;				// 存放各級頻繁項目個數(shù)
	vector<int>	candlargeItemCount;			// 存放各級候選項目個數(shù)

	vector< ItemType >	freqGroup;			// 存放所有級別的頻繁項集

//函數(shù)
public:
	CAprioriSet(int _itemCount, double _itemSupp);

	~CAprioriSet(void);
	// 查找1-頻繁項集
	void FindlargeItem(void);	
	// 生成頻繁項集，作為算法結(jié)果返回給view
	void GenfreqItem(int level);
	// 剪枝函數(shù)
	bool Prune( int level, string& strCandFreqItem, vector<bool> itemFlag );
	// 抽取字符串中的項目
	vector<string>& tokenItem(vector<string>& strVec , string& text);
	// 由L(k-1)生成C(k)
	void AprioriGen(int level);
	// 獲取頻繁項集，結(jié)果保存為vector <tuple<k, item, count> >
	vector< ItemType > getFreqItem();
	// 生成下一個組合
	string& next_combination(vector<string>& kItem, int vecPos);
	// 搜索數(shù)據(jù)集中的子串
	bool search(vector<string>& src, string& text);
};