2659: [Beijing wc2012]算不出的算式

2659: [Beijing wc2012]算不出的算式

Time Limit: 3 Sec  Memory Limit: 128 MB
Submit: 1208  Solved: 711
[ Submit][ Status][ Discuss]

Description

算不出的算式
背景：
曾经有一个老掉牙的游戏放在我面前，我没有珍惜。直到这个游戏停产才追悔莫及。人世间最痛苦的事情莫过于此，如果上天给我一个再玩一次的机会，我一定要，通关！
题目描述：
如果你真的很想玩这个游戏，那么就先看看我的题目吧，搞不定这些的话是没办法通关的哟。第一关其实很简单，只有一个关闭的有密码锁的大门。这大门上写着一个奇怪的算式，估计是要你利用它算出密码来开门吧（果然是老掉牙的情节）。
传说中这个式子中的p和q是两个奇质数，等号右边算出来应该就是密码了吧，你是真的算不出来么？

Input

只有一行，两个奇质数，分别表示p,q。

Output

一个数,表示算式结果。

Sample Input

5 7

Sample Output

6

HINT

HINT:p,q在32位整型范围内。

Source

An Introduction to the Theory of Numbers第6版定理100

[ Submit][ Status][ Discuss]
两个求和分别看做是两条线段求其下方的格点数 套用模板解决--(只会背公式的苟蒻) 对于 解决步骤为，，， 先将a,d化简为小于m，多出的部分O(1)算出，然后套用公式

l = (a + dn) / m

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<bitset>
#include<algorithm>
#include<cstring>
#include<map>
#include<stack>
#include<set>
#include<cmath>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;

typedef long long LL;

LL p,q;

LL Solve(LL a,LL d,LL m,LL n)
{
	LL ret = 0;
	if (a >= m) {
		ret += (a/m)*n;
		a %= m;
	}
	if (d >= m) {
		ret += (d/m)*(n)*(n-1)/2LL;
		d %= m;
	}
	if (!d) {
		ret += (a/m)*n;
		return ret;
	}
	return ret + Solve((a + d*n)%m,m,d,(a + d*n)/m);
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> p >> q;
	cout << Solve(0,q,p,p/2 + 1) + Solve(0,p,q,q/2 + 1);
	return 0;
}