STL的rotate其实就可以看做是一个循环移位的函数。

    首先对于循环移位操作有以下的几种思路:

1)最简单的想法就是每次移动一位进行循环遍历整个容器,算法复杂度为O(n*m)。

2)运用分组交换(尽可能使数组的前面连续几个数为所要结果):若a长度大于b,将ab分成a0a1b,交换a0和b,得ba1a0,只需再迭代交换a1 和a0。算法复杂度为O(n)。

3)运用3次翻转操作(reverse),因为(arbr)r=ab。算法复杂度为O(n)。

4)构建循环操作链,对链中的元素进行循环移位。算法复杂度为O(n)。

    这里链数为gcd(m,n) 。链中元素个数为n/gcd(m,n)。

注:方案4涉及到数论中的一个小定理:若有两个正整数m、n,且gcd(m,n)=d,那么序列{m%n, 2m%n, 3m%n,..., nm%n}一定是{0, d, 2d,..., n-d}的某个排列并重复出现d次,其中%号代表求模操作。比如若m=6, n=8,d=gcd(m,n)=2,那么{6%8, 12%8, 18%8,..., 48%8}即为{0,2,4,6}的某个排列并重复。特别地,若m、n互素,d=1,那么序列{m%n,2m%n,3m%n,...,(n-1)m%n}实际上就是{1, 2, 3,..., n-1}的某个排列。这个定理的证明过程可以很多书中找到(比如具体数学4.8节),这里不再详述。

    

    对于STL的rotate函数针对不同类型的迭代器有不同的效率,共分为3个版本(ForwardIterator,BidirectionalIterator,RandomAccessIterator)。

由于BidirectionIterator有双端移动的能力,所以能调用reverse函数。选择方案3。

由于RandomAccessIterator具有+n操作能力,所以选择方案4。

 

现在的问题是,后三种算法的复杂度都是O(n),那么为什么要用三种不同的算法??

// 分派函数(dispatch function)
template <class BidirectionalIterator>
inline void reverse(BidirectionalIterator first, BidirectionalIterator last) {
  __reverse(first, last, iterator_category(first));
}

// reverse 的 bidirectional iterator 版
template <class BidirectionalIterator>
void __reverse(BidirectionalIterator first, BidirectionalIterator last, 
               bidirectional_iterator_tag) {
  while (true)
    if (first == last || first == --last)   //剩余需交换元素个数小于2
      return;
    else
      iter_swap(first++, last);             //iter_swap()函数为交换迭代器所指元素的值
}

// reverse 的 random access iterator 版
template <class RandomAccessIterator>
void __reverse(RandomAccessIterator first, RandomAccessIterator last,
               random_access_iterator_tag) {
  // 注意,以下的 first < last 判断动作,只适用于 random iterators. 
  while (first < last) iter_swap(first++, --last); 
}

//_copy版本的revese函数
template <class BidirectionalIterator, class OutputIterator>
OutputIterator reverse_copy(BidirectionalIterator first,
                            BidirectionalIterator last,
                            OutputIterator result) {
  while (first != last) {	// 遍历整个序列
    --last;				     
    *result = *last;		
    ++result;				  }
  return result;
}

// rotate 的 forward iterator 版
template <class ForwardIterator, class Distance>
void __rotate(ForwardIterator first, ForwardIterator middle,
              ForwardIterator last, Distance*, forward_iterator_tag) {
  for (ForwardIterator i = middle; ;) {
    iter_swap(first, i);	// 前后段元素一一交换
    ++first;				// 双双前进1
    ++i;					
    // 以下判断是前段[first, middle)先结束还是后段[middle,last)先结束
    if (first == middle) {		// 前段先结束
      if (i == last) return; 	// 如果后段也结束了,那么就结束了
      middle = i;				// 否则进行调整,之后再进行迭代
    }
    else if (i == last)	     // 后段先结束
      i = middle;			// 否则进行调整,之后再进行迭代
  }
}

// rotate 的 bidirectional iterator 版
template <class BidirectionalIterator, class Distance>
void __rotate(BidirectionalIterator first, BidirectionalIterator middle,
              BidirectionalIterator last, Distance*,
              bidirectional_iterator_tag) {
  reverse(first, middle);
  reverse(middle, last);
  reverse(first, last);
}

// 最大公因数,利用辗转相除法
// __gcd() 应用于 __rotate() 的 random access iterator 版
template <class EuclideanRingElement>
EuclideanRingElement __gcd(EuclideanRingElement m, EuclideanRingElement n)
{
  while (n != 0) {           //这里是一个迭代
    EuclideanRingElement t = m % n;
    m = n;
    n = t;
  }
  return m;
}

// rotate 的 random access iterator 版
template <class RandomAccessIterator, class Distance>
void __rotate(RandomAccessIterator first, RandomAccessIterator middle,
              RandomAccessIterator last, Distance*,
              random_access_iterator_tag) {
  // 以下迭代器的相减操作只使用于RandomAccessIterator
  // 取全长和前段(移动的位数)的最大公因数
Distance n = __gcd(last - first, middle - first);
// 链数为gcd(m,n) 。链中元素个数为n/gcd(m,n)。
  while (n--)  //为了书写方便,先从最后一条链开始循环                                               
    __rotate_cycle(first, last, first + n, middle - first,
                   value_type(first));
}

template <class RandomAccessIterator, class Distance, class T>
void __rotate_cycle(RandomAccessIterator first, RandomAccessIterator last,
                    RandomAccessIterator initial, Distance shift, T*) {
  T value = *initial;                 //记下链首元素的值,接下来链首元素“出列”留下一个“槽”
  RandomAccessIterator ptr1 = initial;
  RandomAccessIterator ptr2 = ptr1 + shift;//指向链中下一元素
  while (ptr2 != initial) {
    *ptr1 = *ptr2;
    ptr1 = ptr2;                  //ptr1指向“槽”的位置
    if (last - ptr2 > shift)   //还没有到达最后一个元素
      ptr2 += shift;
else
//为了跳出循环!真有点不明白为什么不用while(true),然后此处直接break??
      ptr2 = first + (shift - (last - ptr2)); 
  }
  *ptr1 = value;
}

// 分派函式(dispatch function)
template <class ForwardIterator>
inline void rotate(ForwardIterator first, ForwardIterator middle,
                   ForwardIterator last) {
  if (first == middle || middle == last) return;         //交换的中间点在首尾,返回
  __rotate(first, middle, last, distance_type(first),
           iterator_category(first));
}

//_copy版本的rotate
template <class ForwardIterator, class OutputIterator>
OutputIterator rotate_copy(ForwardIterator first, ForwardIterator middle,
                           ForwardIterator last, OutputIterator result) {
  return copy(first, middle, copy(middle, last, result));
}


 

    现在可以进行之前问题的分析了,由于ForwardIterator版本需要进行迭代,所以引入了更多的判断(前后段的结束次序)。

BidirectionalIterator版本不需要进行迭代进行n次交换就可以了。RandomAccessIterator版本与BidirectionalIterator版本最大的区别是交换方式。B版本的交换是使用两两交换(iter_swap),R版本的交换是使用循环赋值。所以B版本操作一次需要3次

赋值操作,而R版本只需要一次。也可以说B版本复杂度为O(3tn),R版本复杂度为O(tn)。

 

 

 

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐