Update index.go

Author: masx200

er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/index.go

@@ -37,6 +37,14 @@ func ReverseInOrderIterator(ctx context.Context, root *TreeNode) (gen chan int)
 
 			return
 		}
+
+select {
+		case <-ctx.Done():
+			return
+		default:
+			target <- v
+		}
+
 		YieldAll(ctx, gen, ReverseInOrderIterator(ctx, root.Right))
 		YieldOne(
 			ctx, gen, root.Val)