当然可以实现！用LaTeX的listings包完全能手动控制代码片段的颜色，刚好满足你跟踪项重写过程中代码移位的可视化需求。下面给你两种简单实用的方法：

方法一：定义自定义颜色切换命令 #

这种方式更清晰，适合频繁切换颜色的场景：

首先在导言区导入必要的包并定义命令：

\usepackage{listings}
\usepackage{xcolor} % 提供颜色支持

% 自定义颜色切换命令
\newcommand{\turnGreen}{\color{green}}
\newcommand{\turnRed}{\color{red}}
\newcommand{\resetColor}{\color{black}} % 恢复默认颜色，可根据你的基础样式调整

% 配置listings，开启LaTeX命令逃逸功能
\lstset{
    language=Haskell,
    escapeinside={/*@}{@*/}, % 定义逃逸标记，选Haskell代码里不会出现的组合
    basicstyle=\ttfamily, % 基础代码样式，可保留你原本的设置
    % 其他你需要的配置：比如 frame=single, numbers=left 等
}

然后在代码块中插入命令切换颜色：

\begin{lstlisting}
-- 默认黑色代码
map f [] = []
/*@\turnGreen@*/
-- 从这里开始变为绿色，标记重写的起始片段
map f (x:xs) = f x : map f xs
/*@\turnRed@*/
-- 切换为红色，标记移位后的片段
map (+1) [1,2,3] = 2 : map (+1) [2,3]
/*@\resetColor@*/
-- 恢复默认黑色
map (+1) [2,3] = 3 : map (+1) [3]
\end{lstlisting}

方法二：直接插入颜色命令（无需自定义） #

如果只是偶尔切换颜色，也可以直接在代码中插入\color命令，同样需要开启逃逸功能：

导言区配置：

\usepackage{listings}
\usepackage{xcolor}

\lstset{
    language=Haskell,
    escapeinside={/*@}{@*/}, % 同样需要逃逸标记
    basicstyle=\ttfamily,
}

代码块中直接使用：

\begin{lstlisting}
foldr f z [] = z
/*@\color{green}@*/
foldr f z (x:xs) = f x (foldr f z xs)
/*@\color{red}@*/
foldr (+) 0 [1,2,3] = 1 + foldr (+) 0 [2,3]
/*@\color{black}@*/
foldr (+) 0 [2,3] = 2 + foldr (+) 0 [3]
\end{lstlisting}

注意事项 #

• 逃逸标记（比如/*@和@*/）要选你的Haskell代码中不会出现的字符组合，避免和代码本身冲突，比如可以换成<!!>和<!!>这类符号。
• 如果你的basicstyle原本设置了默认颜色（比如\ttfamily\color{darkgray}），记得把resetColor的颜色改成对应的默认值，确保能正确恢复。
• 这种手动切换的方式完全灵活，你可以在任意位置（行内、行首、行中）切换颜色，完美适配项重写过程中跟踪代码移位的需求。

内容的提问来源于stack exchange，提问作者Raoul