Для заданных двух целых чисел A и B найдите пару чисел X и Y такую, что A = X * Y и B = X xor Y

Я борюсь с этой проблемой, которую нашел в конкурентной книге по программированию, но без решения, как это сделать.

Для заданных двух целых чисел A и B (может соответствовать 64-разрядному целочисленному типу), где A нечетно, найдите пару чисел X и Y, такую, что A = X * Y и B = X или Y. Мой подход заключался в том, чтобы перечислить все делители а и попробуйте спаривания номера под SQRT (а) с номерами над SQRT (A) , которые размножаются до а и посмотреть , если их исключающее равно B . Но я не знаю, достаточно ли это эффективно. Каково было бы хорошее решение / алгоритм для этой проблемы?

Вот простая рекурсия, которая соблюдает известные нам правила: (1) устанавливаются младшие значащие биты как X, так и Y, поскольку только нечетные мультипликаторы дают нечетное кратное число; (2) если мы установим для X самый высокий установленный бит B, Y не может быть больше sqrt (A); и (3) установить биты в X или Y в соответствии с текущим битом в B.

Следующий код Python дал менее 300 итераций для всех, кроме одной случайной пары, которую я выбрал из примера кода Мэтта Тиммерманса . Но первый занял 231 199 итераций :)

from math import sqrt

def f(A, B):
  i = 64
  while not ((1<<i) & B):
    i = i - 1
  X = 1 | (1 << i)

  sqrtA = int(sqrt(A))

  j = 64
  while not ((1<<j) & sqrtA):
    j = j - 1

  if (j > i):
    i = j + 1

  memo = {"it": 0, "stop": False, "solution": []}

  def g(b, x, y):
    memo["it"] = memo["it"] + 1
    if memo["stop"]:
      return []

    if y > sqrtA or y * x > A:
      return []

    if b == 0:
      if x * y == A:
        memo["solution"].append((x, y))
        memo["stop"] = True
        return [(x, y)]
      else:
        return []

    bit = 1 << b

    if B & bit:
      return g(b - 1, x, y | bit) + g(b - 1, x | bit, y)
    else:
      return g(b - 1, x | bit, y | bit) + g(b - 1, x, y)

  g(i - 1, X, 1)
  return memo

vals = [
  (6872997084689100999, 2637233646), # 1048 checks with Matt's code
  (3461781732514363153, 262193934464), # 8756 checks with Matt's code
  (931590259044275343, 5343859294), # 4628 checks with Matt's code
  (2390503072583010999, 22219728382), # 5188 checks with Matt's code
  (412975927819062465, 9399702487040), # 8324 checks with Matt's code
  (9105477787064988985, 211755297373604352), # 3204 checks with Matt's code
  (4978113409908739575,67966612030), # 5232 checks with Matt's code
  (6175356111962773143,1264664368613886), # 3756 checks with Matt's code
  (648518352783802375, 6) # B smaller than sqrt(A)
]

for A, B in vals:
  memo = f(A, B)
  [(x, y)] = memo["solution"]
  print "x, y: %s, %s" % (x, y)
  print "A:   %s" % A
  print "x*y: %s" % (x * y)
  print "B:   %s" % B
  print "x^y: %s" % (x ^ y)
  print "%s iterations" % memo["it"]
  print ""

Вывод:

x, y: 4251585939, 1616572541
A:   6872997084689100999
x*y: 6872997084689100999
B:   2637233646
x^y: 2637233646
231199 iterations

x, y: 262180735447, 13203799
A:   3461781732514363153
x*y: 3461781732514363153
B:   262193934464
x^y: 262193934464
73 iterations

x, y: 5171068311, 180154313
A:   931590259044275343
x*y: 931590259044275343
B:   5343859294
x^y: 5343859294
257 iterations

x, y: 22180179939, 107776541
A:   2390503072583010999
x*y: 2390503072583010999
B:   22219728382
x^y: 22219728382
67 iterations

x, y: 9399702465439, 43935
A:   412975927819062465
x*y: 412975927819062465
B:   9399702487040
x^y: 9399702487040
85 iterations

x, y: 211755297373604395, 43
A:   9105477787064988985
x*y: 9105477787064988985
B:   211755297373604352
x^y: 211755297373604352
113 iterations

x, y: 68039759325, 73164771
A:   4978113409908739575
x*y: 4978113409908739575
B:   67966612030
x^y: 67966612030
69 iterations

x, y: 1264664368618221, 4883
A:   6175356111962773143
x*y: 6175356111962773143
B:   1264664368613886
x^y: 1264664368613886
99 iterations

x, y: 805306375, 805306369
A:   648518352783802375
x*y: 648518352783802375
B:   6
x^y: 6
59 iterations

Вы знаете, что по крайней мере один фактор <= sqrt (A). Давайте сделаем это один X.

Длина X в битах будет примерно вдвое меньше длины A.

Следовательно, все старшие биты X - те, которые имеют более высокое значение, чем sqrt (A), - равны 0, и соответствующие биты в B должны иметь то же значение, что и соответствующие биты в Y.

Знание старших битов Y дает вам довольно маленький диапазон для соответствующего коэффициента X = A / Y. Вычислите Xmin и Xmax, соответствующие наибольшим и наименьшим возможным значениям для Y соответственно. Помните, что Xmax также должен быть <= sqrt (A).

Затем просто попробуйте все возможные X между Xmin и Xmax. Там не будет слишком много, поэтому это не займет много времени.

Другой простой способ решения этой проблемы заключается в том, что младшие n битов XY и X xor Y зависят только от младших n битов X и Y. Поэтому вы можете использовать возможные ответы для младших n битов, чтобы ограничить возможные ответы для младших n + 1 бит, пока вы не закончите.

Я понял, что, к сожалению, может быть более одной возможности для одного n . Я не знаю, как часто будет много возможностей, но это, вероятно, не так уж часто, если вообще так, так что это может быть хорошо в конкурентном контексте. Вероятно, будет только несколько возможностей, поскольку решение для n битов обеспечит либо 0, либо два решения для n + 1 битов с равной вероятностью.

Кажется, это работает довольно хорошо для случайного ввода. Вот код, который я использовал для его проверки:

public static void solve(long A, long B)
{
    List<Long> sols = new ArrayList<>();
    List<Long> prevSols = new ArrayList<>();
    sols.add(0L);
    long tests=0;
    System.out.print("Solving "+A+","+B+"... ");
    for (long bit=1; (A/bit)>=bit; bit<<=1)
    {
        tests += sols.size();
        {
            List<Long> t = prevSols;
            prevSols = sols;
            sols = t;
        }
        final long mask = bit|(bit-1);
        sols.clear();
        for (long prevx : prevSols)
        {
            long prevy = (prevx^B) & mask;
            if ((((prevx*prevy)^A)&mask) == 0)
            {
                sols.add(prevx);
            }
            long x = prevx | bit;
            long y = (x^B)&mask;
            if ((((x*y)^A)&mask) == 0)
            {
                sols.add(x);
            }
        }
    }
    tests += sols.size();
    {
        List<Long> t = prevSols;
        prevSols = sols;
        sols = t;
    }
    sols.clear();
    for (long testx: prevSols)
    {
        if (A/testx >= testx)
        {
            long testy = B^testx;
            if (testx * testy == A)
            {
                sols.add(testx);
            }
        }
    }

    System.out.println("" + tests + " checks -> X=" + sols);
}
public static void main(String[] args)
{
    Random rand = new Random();
    for (int range=Integer.MAX_VALUE; range > 32; range -= (range>>5))
    {
        long A = rand.nextLong() & Long.MAX_VALUE;
        long X = (rand.nextInt(range)) + 2L;
        X|=1;
        long Y = A/X;
        if (Y==0)
        {
            Y = rand.nextInt(65536);
        }
        Y|=1;
        solve(X*Y, X^Y);
    }
}

Вы можете увидеть результаты здесь: https://ideone.com/cEuHkQ

Похоже, что обычно требуется всего несколько тысяч проверок.