Update grover

itsubaki · web-flow · commit 5f1786d48feb · 2026-01-25T00:46:12.000+09:00
diff --git a/cmd/grover/main.go b/cmd/grover/main.go
@@ -62,10 +62,10 @@ func oracle(qsim *q.Q, r, s []q.Qubit, a q.Qubit) {
 	xor(r[0], r[1], s[0])
 }
 
-func diffuser(qsim *q.Q, r []q.Qubit) {
+func diffuser(qsim *q.Q, r []q.Qubit, a q.Qubit) {
 	qsim.H(r...)
 	qsim.X(r...)
-	qsim.ControlledZ(r[:len(r)-1], []q.Qubit{r[len(r)-1]})
+	qsim.ControlledZ(r, []q.Qubit{a})
 	qsim.X(r...)
 	qsim.H(r...)
 }
@@ -95,7 +95,7 @@ func main() {
 	// iterations
 	for range R {
 		oracle(qsim, r, s, a)
-		diffuser(qsim, r)
+		diffuser(qsim, r, a)
 	}
 
 	// quantum states

Original file line number	Diff line number	Diff line change
`@@ -62,10 +62,10 @@ func oracle(qsim *q.Q, r, s []q.Qubit, a q.Qubit) {`
`62`	`62`	`xor(r[0], r[1], s[0])`
`63`	`63`	`}`
`64`	`64`
`65`		`-func diffuser(qsim *q.Q, r []q.Qubit) {`
	`65`	`+func diffuser(qsim *q.Q, r []q.Qubit, a q.Qubit) {`
`66`	`66`	`qsim.H(r...)`
`67`	`67`	`qsim.X(r...)`
`68`		`- qsim.ControlledZ(r[:len(r)-1], []q.Qubit{r[len(r)-1]})`
	`68`	`+ qsim.ControlledZ(r, []q.Qubit{a})`
`69`	`69`	`qsim.X(r...)`
`70`	`70`	`qsim.H(r...)`
`71`	`71`	`}`
`@@ -95,7 +95,7 @@ func main() {`
`95`	`95`	`// iterations`
`96`	`96`	`for range R {`
`97`	`97`	`oracle(qsim, r, s, a)`
`98`		`- diffuser(qsim, r)`
	`98`	`+ diffuser(qsim, r, a)`
`99`	`99`	`}`
`100`	`100`
`101`	`101`	`// quantum states`