Dynamische Programmierung mit vielen Teilproblemen. Ich versuche also, dieses Problem in der Interview Street zu lösen:
Gitter Gehen (50 Punkte) ,
Sie sind in einem befindet -dimensionalen Gitter an der Position . Die Abmessungen des Gitters sind ). In einem Schritt können Sie in einer der Dimensionen einen Schritt vorwärts oder rückwärts gehen. (Es sind also immer verschiedene Bewegungen möglich). Auf wie viele Arten können Sie M nehmenSchritte, bei denen Sie das Raster zu keinem Zeitpunkt verlassen? Sie verlassen das Gitter, wenn für entweder oder .
Mein erster Versuch war diese auswendig gelernte rekursive Lösung:
def number_of_ways(steps, starting_point):
global n, dimensions, mem
#print steps, starting_point
if (steps, tuple(starting_point)) in mem:
return mem[(steps, tuple(starting_point))]
val = 0
if steps == 0:
val = 1
else:
for i in range(0, n):
tuple_copy = starting_point[:]
tuple_copy[i] += 1
if tuple_copy[i] <= dimensions[i]:
val += number_of_ways(steps - 1, tuple_copy)
tuple_copy = starting_point[:]
tuple_copy[i] -= 1
if tuple_copy[i] > 0:
val += number_of_ways(steps - 1, tuple_copy)
mem[(steps, tuple(starting_point))] = val
return val
Große Überraschung: Es schlägt bei einer großen Anzahl von Schritten und / oder Dimensionen aufgrund von Speichermangel fehl.
Der nächste Schritt ist also, meine Lösung durch dynamische Programmierung zu verbessern. Aber bevor ich anfange, sehe ich ein großes Problem mit dem Ansatz. Das Argument starting_point
ist ein Tupel, wobei so groß wie . So in der Tat könnte die Funktion sein , number_of_ways(steps, x1, x2, x3, ... x10)
mit .
Die dynamischen Programmierprobleme, die ich in Lehrbüchern gesehen habe, haben fast alle zwei Variablen, so dass nur eine zweidimensionale Matrix benötigt wird. In diesem Fall wäre eine zehndimensionale Matrix erforderlich. Also Zellen insgesamt.
Bei 2D-Matrizen in der dynamischen Programmierung wird normalerweise nur die vorherige Berechnungsreihe für die nächste Berechnung benötigt, wodurch die räumliche Komplexität von auf min ( m , n ) verringert wird . Ich bin mir nicht sicher, wie ich das in diesem Fall tun würde. Die Visualisierung einer Tabelle ist nicht möglich, daher müsste die Antwort direkt aus der obigen Rekursion stammen.
AKTUALISIEREN
Verwenden Sie die Vorschläge von Peter Shor und nehmen Sie einige kleinere Korrekturen vor, insbesondere die Notwendigkeit, die Position in der -Funktion zu verfolgen und nicht nur die Dimensionen in zwei Sätze A und B aufzuteilen, sondern die Aufteilung rekursiv und effektiv durchzuführen Eine Divide-and-Conquer-Methode, bis ein Basisfall erreicht ist, in dem sich nur eine Dimension in der Menge befindet.
Ich habe die folgende Implementierung entwickelt, die alle Tests unterhalb der maximalen Ausführungszeit bestanden hat:
def ways(di, offset, steps):
global mem, dimensions
if steps in mem[di] and offset in mem[di][steps]:
return mem[di][steps][offset]
val = 0
if steps == 0:
val = 1
else:
if offset - 1 >= 1:
val += ways(di, offset - 1, steps - 1)
if offset + 1 <= dimensions[di]:
val += ways(di, offset + 1, steps - 1)
mem[di][steps][offset] = val
return val
def set_ways(left, right, steps):
# must create t1, t2, t3 .. ti for steps
global mem_set, mem, starting_point
#print left, right
#sleep(2)
if (left, right) in mem_set and steps in mem_set[(left, right)]:
return mem_set[(left, right)][steps]
if right - left == 1:
#print 'getting steps for', left, steps, starting_point[left]
#print 'got ', mem[left][steps][starting_point[left]], 'steps'
return mem[left][steps][starting_point[left]]
#return ways(left, starting_point[left], steps)
val = 0
split_point = left + (right - left) / 2
for i in xrange(steps + 1):
t1 = i
t2 = steps - i
mix_factor = fact[steps] / (fact[t1] * fact[t2])
#print "mix_factor = %d, dimension: %d - %d steps, dimension %d - %d steps" % (mix_factor, left, t1, split_point, t2)
val += mix_factor * set_ways(left, split_point, t1) * set_ways(split_point, right, t2)
mem_set[(left, right)][steps] = val
return val
import sys
from time import sleep, time
fact = {}
fact[0] = 1
start = time()
accum = 1
for k in xrange(1, 300+1):
accum *= k
fact[k] = accum
#print 'fact_time', time() - start
data = sys.stdin.readlines()
num_tests = int(data.pop(0))
for ignore in xrange(0, num_tests):
n_and_steps = data.pop(0)
n, steps = map(lambda x: int(x), n_and_steps.split())
starting_point = map(lambda x: int(x), data.pop(0).split())
dimensions = map(lambda x: int(x), data.pop(0).split())
mem = {}
for di in xrange(n):
mem[di] = {}
for i in xrange(steps + 1):
mem[di][i] = {}
ways(di, starting_point[di], i)
start = time()
#print 'mem vector is done'
mem_set = {}
for i in xrange(n + 1):
for j in xrange(n + 1):
mem_set[(i, j)] = {}
answer = set_ways(0, n, steps)
#print answer
print answer % 1000000007
#print time() - start
mem[]
Wörterbuchs erschöpft wird . Und danke, dass du meine Antwort aufgeräumt hast. Nicht allzu vertraut mit LaTeX, wird sich aber beim nächsten Mal anstrengen.