flavius.cpp

来自「Ulm大学2005-2006年竞赛题」· C++ 代码 · 共 38 行

// Problem   Flavius Josephus
// Algorithm Pollard Rho
// Runtime   O(n)
// Author    Adrian Kuegel
// Date      2006.06.25

#include <stdio.h>
#include <assert.h>

int next(long long cur, long long a, long long b, long long n) {
	return (a*((cur*cur)%n)+b)%n;
}

int main() {
	freopen("flavius.in","r",stdin);
	int n,a,b;
	while(scanf("%d",&n) == 1 && n != 0) {
		assert(n >= 2 && n <= 1000000000);
		assert(scanf("%d %d",&a,&b) == 2);
		assert(a >= 0 && a < n && b >= 0 && b < n);
		int p1 = 0, p2 = 0, len = 0;
		int cnt = 0;
		// find a value in the cycle
		do {
			p1 = next(p1,a,b,n); // advance pointer 1 in steps of 1
			p2 = next(next(p2,a,b,n),a,b,n); // advance pointer 2 in steps of 2
			++cnt;
		}while(p1 != p2);
		// find cycle length
		do {
			p2 = next(p2,a,b,n);
			++len;
		}while(p1 != p2);
		printf("%d\n",n-len);
	}
	return 0;
}