結果

問題 No.442 和と積
コンテスト
ユーザー bal4u
提出日時 2019-04-18 17:33:03
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 2,761 bytes
コンパイル時間 223 ms
コンパイル使用メモリ 33,152 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-04 23:17:46
合計ジャッジ時間 867 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

// yukicoder: No.442 和と積
// 2019.4.17 bal4u

#include <stdio.h>
#include <math.h>

long long gcd(long long a, long long b)
{
	long long r;
	while (b != 0) r = a % b, a = b, b = r;
	return a;
}

#define SIZE  50
long long factor[2][SIZE];
int power[2][SIZE];
int sz[2];

int ptbl[] = {
		 3,   5,   7,  11,  13,  17,  19,  23,  29,
   31,  37,  41,  43,  47,  53,  59,  61,  67,  71,
   73,  79,  83,  89,  97, 101, 103, 107, 109, 113,
  127, 131, 137, 139, 149, 151, 157, 163, 167, 173,
  179, 181, 191, 193, 197, 199, 211, 223, 227, 229,
  233, 239, 241, 251, 257, 263, 269, 271, 277, 281,
  283, 293, 307, 311, 313, 317, 331, 337, 347, 349,
  353, 359, 367, 373, 379, 383, 389, 397, 401, 409,
  419, 421, 431, 433, 439, 443, 449, 457, 461, 463,
  467, 479, 487, 491, 499, 503, 509, 521, 523, 541,
  547, 557, 563, 569, 571, 577, 587, 593, 599, 601,
  607, 613, 617, 619, 631, 641, 643, 647, 653, 659,
  661, 673, 677, 683, 691, 701, 709, 719, 727, 733,
  739, 743, 751, 757, 761, 769, 773, 787, 797, 809,
  811, 821, 823, 827, 829, 839, 853, 857, 859, 863,
  877, 881, 883, 887, 907, 911, 919, 929, 937, 941,
  947, 953, 967, 971, 977, 983, 991, 997, 0 };

int prime_factor(int id, long long n)
{
	int i, d, size;
	int *pp;

	size = 0;
	if ((n & 1) == 0) {
		factor[id][size] = 2;
		do n >>= 1, power[id][size]++;
		while ((n & 1) == 0);
		size++;
	}
	for (pp = ptbl; n > 1 && *pp > 0; pp++) {
		if (n % *pp) continue;
		d = *pp;
		factor[id][size] = d;
		do n /= d, power[id][size]++;
		while (n % d == 0);
		size++;
	}
	if (n > 1) {
		int b = (int)sqrt((double)n);
		for (i = 1009; n > 1; i += 2) {
			if (i > b) {
				factor[id][size] = n, power[id][size++] = 1;
				break;
			}
			if (n % i == 0) {
				factor[id][size] = i;
				do n /= i, power[id][size]++;
				while (n % i == 0);
				size++;
			}
		}
	}
	return size;
}

int f[SIZE];

int main()
{
	int i, j;
	long long A, B, g1, g2, k, ans;

	scanf("%lld%lld", &A, &B);
	if (A == B) {
		if (A & 1) ans = A;
		else ans = A << 1;
		printf("%lld\n", ans);
		return 0;
	}

	g1 = gcd(A+B, A), g2 = gcd(A+B, B);
	sz[0] = prime_factor(0, g1);
	sz[1] = prime_factor(1, g2);

	i = j = 0; while (i < sz[0] && j < sz[1]) {
		while (factor[1][j] > factor[0][i] && i < sz[0]) i++;
		while (factor[1][j] < factor[0][i] && j < sz[1]) j++;
		if (i < sz[0] && j < sz[1] && factor[0][i] == factor[1][j]) {
			power[0][i] += power[1][j], factor[1][j] = 0;
			i++, j++;
		}
	}
	ans = 1, A += B;
	for (i = 0; i < sz[0]; i++) {
		k = 1; while (power[0][i]--) k *= factor[0][i];
		while (A % k) k /= factor[0][i];
		ans *= k;
	}
	for (i = 0; i < sz[1]; i++) if (factor[1][i] > 0) {
		k = 1; while (power[1][i]--) k *= factor[1][i];
		while (A % k) k /= factor[1][i];
		ans *= k;
	}
	printf("%lld\n", ans);
	return 0;
}
0