再说约瑟夫环

题目来自@陈利人的微博:1到n个人排成一圈,从编号为1的人开始,依次说白,黑,白,黑,。。。,说到黑的人离队,问最后留下编号为几的人。

大家都知道,这是非常典型的约瑟夫环问题。其一般形式是M个人,构成一个环(圆),每隔N人就滚蛋,问最后剩下谁。

很多同学说答案是1,这可能是因为没理解题目(没上过数据结构?)。比如说三个人,A、B、C,编号分别是1、2、3.

A说“白”,B说“黑”,C说“白”,然后B滚蛋了,那么剩下A、C。注意,C已经说”白“了,而大家是构成环的,所以A要说“黑”了,所以A滚蛋。最终剩下编号为3的C。

通过一个循环链表(队列)可以很方便的模拟。代码我就不写了。模拟,是ACM题目训练里的三大常见题型之一:模拟、数论、DP、贪心、字符串、搜索

通过分析(大家可以参考这个链接:http://www.zhihu.com/question/20065611或者参考离散数学教材),可以通过简单的循环左移一位来获得结果,复杂度为O(1)。

比如说n=5,那么二进制表示为101,循环左移一位变成011,也就说如果n=5,那么最终唯一剩下来的编号是3(011)

但是,这里需要特别注意前导0的存在。比如n=5,那么二进制00000101循环移位后得到00001010结果却是10,这明显是不对的。

 

#include<iostream>

using namespace std;

const int BASE = 32;

int getLeadingZeroCounts(unsigned int x)

{

	int n = 1;

	if (x == 0) return -1;

	if ((x >> 16) == 0) { n = n + 16; x = x << 16; }

	if ((x >> 24) == 0) { n = n + 8; x = x << 8; }

	if ((x >> 28) == 0) { n = n + 4; x = x << 4; }

	if ((x >> 30) == 0) { n = n + 2; x = x << 2; }

	n = n - (x >> 31);

	return n;

}

unsigned int get(unsigned int n)

{

	unsigned int temp1 = (1 << (BASE - getLeadingZeroCounts(n)))-1;

	unsigned int temp2= n << 1;

	unsigned int temp3 = temp1 & temp2;

	return temp3 | 1;

}

int main()

{

	unsigned int n;

	while (cin >> n)

	{

		cout <<get(n) << endl;

	}

}

  

说明:

1,我最早知道循环移位可以解决这个问题,是在高中时候所看的《算法设计与分析基础》(Anany Levitin著)这本书。有兴趣朋友可以参考下。

2,我最早知道代码中获得前导0的数量的实现,是在本科图书馆里看的《Hacker's Delight》(Henry S. Warren著)这本书。有兴趣朋友可以参考下。

当然,估计很多编译器和CPU都有搭配类似的指令了,速度当然很快。

你可能感兴趣的:(约瑟夫环)