我的列表是 l[],包含 1000 个整数。在算法中,我从这个列表中选择前 n 个整数。我如何编写 latex 代码来执行此操作。
我可以使用 argmax 运算符从列表或它们对应的索引中选择最大 k 个值吗?
答案1
仅基于 TeX 原语和基本 Plain TeX 宏的解决方案在这里:
\newcount\selected
\def\alist{}
\def\scanlist#1 {%
\ifx;#1\else
\addtoalist\alist{#1}%
\ifnum\selected>0 \advance\selected by-1
\else \remlast\alist \fi
\expandafter\scanlist
\fi
}
\def\remlast#1{\edef#1{\expandafter}\expandafter\remlastA#1{} }
\def\remlastA #1 #2 {%
\if^#2^\else
\edef\alist{\alist#1 }\afterfi{\remlastA #2 }%
\fi
}
\def\addtoalist#1#2{\def\tmp{#2}\edef#1{\expandafter}\expandafter\addtoalistA#1; }
\def\addtoalistA #1 {%
\ifx;#1\edef\alist{\alist \tmp\space}\else
\ifnum\tmp>#1 \edef\alist{\alist\tmp\space#1 }%
\afterfi{\afterfi{\addtoalistB}}%
\else \edef\alist{\alist #1 }%
\afterfi{\afterfi{\addtoalistA}}%
\fi
\fi
}
\def\addtoalistB#1; {\edef\alist{\alist#1}}
\def\afterfi#1#2\fi{\fi#1}
%% test:
\def\list{12 18 32 127 123 432 456 456 890 543 654 234 12 18 87 96 }
\selected=5 % number of selected numbers
\expandafter\scanlist\list; % does the job, result is in \alist
\meaning\alist % macro-> 800 654 543 456 456
\bye
答案2
我知道您正在使用标签pdftex
,但为了以防万一,这里是使用 LuaLaTeX 的解决方案:
\documentclass{article}
\usepackage{luacode}
\directlua{
function sortf(x, y)
return x > y
end
function maxlist(L, n)
table.sort(L, sortf)
local M = {}
for i = 1, n do
M[i] = L[i]
end
return M
end
a = {1, 3, 2, 7, 4, 0}
n = 3
}
\begin{document}
The list is \directlua{tex.print(a)} and its \directlua{tex.print(n)} biggest elements are \directlua{tex.print(maxlist(a, n))}.
\end{document}
简而言之:
- 您可以在命令中的 LaTeX 中的任何位置插入 Lua 代码
\directlua
。 - 所有
\directlua
块都是“链接的”,因此您可以重复使用一个块中的内容。 - 该
tex.print
函数对于 Lua 将结果发送到 LaTeX 是必需的。
我对数组进行排序以提取最大元素的方式并不是最优雅的方法,但你明白我的意思。
答案3
以下是选择n以逗号分隔的列表中的最大整数。
\documentclass{article}
\ExplSyntaxOn
\NewDocumentCommand{\listmax}{mm}{ % #1: list, #2: n
\clist_set:Nn \l_tmpb_clist {#1}
\clist_sort:Nn \l_tmpb_clist
{ \int_compare:nTF { ##1 < ##2 } { \sort_return_swapped: } { \sort_return_same: } }
\clist_clear:N \l_tmpa_clist
\int_step_inline:nn { #2 } { \clist_put_right:Nn \l_tmpa_clist {\clist_item:Nn \l_tmpb_clist { ##1 }} }
\clist_use:Nn \l_tmpa_clist {,~}
}
\ExplSyntaxOff
\begin{document}
\listmax{2,11,13,8,4,1,3,5,7,9,20,6,101}{10}
\listmax{1,6,4,7,8,9}{4}
\end{document}