首页 > 代码库 > 利用stl的集合类函数 操作mfc数组

利用stl的集合类函数 操作mfc数组

#include  <algorithm>
#include  "afxtempl.h"
//TYPE类必须有<
template<class TYPE, class ARG_TYPE>
void Sort(CArray<TYPE,ARG_TYPE>& dest)
{
 std::sort(dest.GetData(),dest.GetData()+dest.GetSize());
}
//构造一个有序序列,该序列仅保留第一个序列中存在的而第二个中不存在的元素。
//src1,src2必须已排序
template<class TYPE, class ARG_TYPE>
void SetDifference(CArray<TYPE,ARG_TYPE>& dest,const CArray<TYPE,ARG_TYPE>& src1,const CArray<TYPE,ARG_TYPE>& src2)
{
 dest.SetSize(src1.GetSize());
 const TYPE* p = std::set_difference(src1.GetData(),src1.GetData()+src1.GetSize(),
   src2.GetData(),src2.GetData()+src2.GetSize(),
   dest.GetData());
 dest.SetSize(p-dest.GetData());
};
//构造一个有序序列,包含两个序列中所有的不重复元素。
//src1,src2必须已排序
template<class TYPE, class ARG_TYPE>
void SetUnion(CArray<TYPE,ARG_TYPE>& dest,const CArray<TYPE,ARG_TYPE>& src1,const CArray<TYPE,ARG_TYPE>& src2)
{
 dest.SetSize(src1.GetSize()+src2.GetSize());
 const TYPE* p = std::set_union(src1.GetData(),src1.GetData()+src1.GetSize(),
   src2.GetData(),src2.GetData()+src2.GetSize(),
   dest.GetData());
 dest.SetSize(p-dest.GetData());
};
//构造一个有序序列,其中元素在两个序列中都存在。
//src1,src2必须已排序
template<class TYPE, class ARG_TYPE>
void SetIntersection(CArray<TYPE,ARG_TYPE>& dest,const CArray<TYPE,ARG_TYPE>& src1,const CArray<TYPE,ARG_TYPE>& src2)
{
 dest.SetSize(min(src1.GetSize(),src2.GetSize()));
 const TYPE* p = std::set_intersection(src1.GetData(),src1.GetData()+src1.GetSize(),
   src2.GetData(),src2.GetData()+src2.GetSize(),
   dest.GetData());
 dest.SetSize(p-dest.GetData());
};
//构造一个有序序列,该序列取两个序列的对称差集(并集-交集)
//src1,src2必须已排序
template<class TYPE, class ARG_TYPE>
void SetSymmetricDifference(CArray<TYPE,ARG_TYPE>& dest,const CArray<TYPE,ARG_TYPE>& src1,const CArray<TYPE,ARG_TYPE>& src2)
{
 dest.SetSize(src1.GetSize()+src2.GetSize());
 const TYPE* p = std::set_symmetric_difference(src1.GetData(),src1.GetData()+src1.GetSize(),
   src2.GetData(),src2.GetData()+src2.GetSize(),
   dest.GetData());
 dest.SetSize(p-dest.GetData());
};
 
CString LngArrToStr(const CArray<long,const long&>& src)//将long数组转成str,方便排错
{
 CString str;
 str.Format("%d个元素 ",src.GetSize());
 for(int i = 0 ; i < src.GetSize() ; i++ )
 {
  CString s;
  s.Format("%d ",src[i]);
  str += s;
 }
 return str;
}
void GongNengCeShi()//功能测试
{
 srand(GetTickCount());
    const int iNum = 11;
 CArray<long,const long&> src1,src2;
 src1.SetSize(iNum);
 src2.SetSize(iNum);
 for( int i = 0 ; i < iNum ; i++)
 {
  src1[i] = rand()%10;
  src2[i] = rand()%10;
 }
 const CString str1 = LngArrToStr(src1);
 const CString str2 = LngArrToStr(src2);
 Sort(src1);
 Sort(src2);
 const CString str3 = LngArrToStr(src1);
 const CString str4 = LngArrToStr(src2);
 CArray<long,const long&> dst;
 SetDifference(dst,src1,src2);
 const CString str5 = LngArrToStr(dst);
 SetUnion(dst,src1,src2);
 const CString str6 = LngArrToStr(dst);
 
 SetIntersection(dst,src1,src2);
 const CString str7 = LngArrToStr(dst);
 SetSymmetricDifference(dst,src1,src2);
 const CString str8 = LngArrToStr(dst);
}
//给数组增加元素,可以连续调用。
template<class TYPE, class ARG_TYPE>
class CTestArray : public CArray<TYPE,ARG_TYPE>
{
public:
 CTestArray<TYPE,ARG_TYPE>& AddLianXu(ARG_TYPE newElement)
 {
  Add(newElement);
  return *this;
 };
};
void BianJieCheShi()//边界测试
{
 CTestArray<long,const long&> src1,src2;
 src1.AddLianXu(1).AddLianXu(2).AddLianXu(3);
 Sort(src1);
 Sort(src2);
 CArray<long,const long&> dst;
 //结果就是src1
 SetDifference(dst,src1,src2);
 const CString str1 = LngArrToStr(dst);
 //结果为空
 SetDifference(dst,src1,src1);
 const CString str2 = LngArrToStr(dst);
 CTestArray<long,const long&> src3;
 src3.AddLianXu(4).AddLianXu(5);
 //无重无元素
 SetUnion(dst,src1,src3);
 const CString str3 = LngArrToStr(dst);
 //全部是重复元素
 SetUnion(dst,src1,src1);
 const CString str4 = LngArrToStr(dst);
 //无重无元素
 SetIntersection(dst,src1,src3);
 const CString str5 = LngArrToStr(dst);
 //全部是重复元素
 SetIntersection(dst,src1,src1);
 const CString str6 = LngArrToStr(dst);
 //无重无元素
 SetSymmetricDifference(dst,src1,src3);
 const CString str7 = LngArrToStr(dst);
 //全部是重复元素
 SetSymmetricDifference(dst,src1,src1);
 const CString str8 = LngArrToStr(dst);
}

利用stl的集合类函数 操作mfc数组