2017年第0届浙江工业大学之江学院程序设计竞赛决赛—A qwb与支教

Description

qwb同时也是是之江学院的志愿者，暑期要前往周边地区支教，为了提高小学生的数学水平。她把小学生排成一排，从左至右从1开始依次往上报数。

玩完一轮后，他发现这个游戏太简单了。于是他选了3个不同的数x,y,z；从1依次往上开始报数，遇到x的倍数、y的倍数或z的倍数就跳过。如果x=2,y=3,z=5；第一名小学生报1，第2名得跳过2、3、4、5、6，报7；第3名得跳过8、9、10，报11。

那么问题来了，请你来计算，第N名学生报的数字是多少？

Input

多组测试数据，处理到文件结束。（测试数据数量<=8000）

每个测试例一行，每行有四个整数x,y,z,N。（ 2≤x,y,z≤10⁷,1≤N≤10¹⁷）。

Output

对于每个测试例，输出第N名学生所报的数字，每个报数占一行。

Sample Input

2 3 5 2
6 2 4 10000

Sample Output

7

思路：简单的容斥原理+二分.

本题的问题可以转化：已知一个数k，另g表示小于等于k的数且满足或者是x的倍数、或者是y的倍数、或者是z的倍数的个数，求第一个数k满足（k-g）== n。k即为所求。

不过我感觉该题的第n个数应该会超longlong范围的，竟然没超..

Code:

#include <bits/stdc++.h>
#define LL long long
using namespace std;
LL x, y, z, n, num[5];
LL gcd(LL a, LL b)
{
	if(a%b == 0) return b;
	return gcd(b, a%b);
}
int calc(LL g)
{
	LL lcm, ans = 0;
	int pos, t, k, up, m = 1;
	if(x <= g) num[m++] = x;
	if(y <= g) num[m++] = y;
	if(z <= g) num[m++] = z;
	--m; up = 1<<m;
	for(int i = 1; i < up; ++i)
	{
		t = i, pos = 1, lcm = 1ll, k = 0;
		while(t)
		{
			if(t&1)
			{
				++k;
				lcm = lcm/gcd(lcm, num[pos])*num[pos];
			}
			++pos; t >>= 1;
		}
		if(k&1) ans += g/lcm;
		else ans -= g/lcm;
	}
	if(g-ans < n) return 0;
	else return 1;
}
int main()
{
	while(~scanf("%lld %lld %lld %lld", &x, &y, &z, &n))
	{
		LL mid, l = 1, r = ~(1ll<<63);
		while(l <= r)
		{
			mid = (l+r)/2;
			if(calc(mid)) r = mid-1;
			else l = mid+1;
		}
		printf("%lld\n", l);
	}
	return 0;
}

小知识补充：(1ll<<63)等于long long能表示的最小负整数，~(1ll<<63)等于long long能表示的最大正整数。

继续加油~