最简单/最快的方法来说明密码学中的安全实验/减少证明

Question

我有三个需要绘制的案例（安全实验和减少证明）：

我正在学习计算机科学，我想为我的考试创建一本“手册”。问题：该过程涉及 30-50 个安全实验 + 安全证明。我需要一种快速简便的方法来创建它们（希望使用 LaTeX）。我希望你能帮助我。

它们不必像我在这里展示的那样 1:1 地呈现；这三个只是我正在寻找的三个例子。我不需要花哨的颜色和有趣的盒子颜色。我更喜欢像第一张图片那样的东西：简单而优雅。

最小工作示例：

\documentclass{article}
\usepackage{amsmath}  
\begin{document}  
\begin{equation}
(PK, TD) \leftarrow Gen(1^n)
\end{equation}

\begin{equation}
x \leftarrow^r x
\end{equation}

\begin{equation}
y \leftarrow f_{PK}(x)
\end{equation}

\begin{equation}
if x = x' \; return \; 1
\end{equation}

\begin{equation}
else \; return \; 0
\end{equation}
\end{document}  

我不知道如何使用加密代码。或者我应该使用 tikz 吗？