
我的列表是包含 1000 個整數的 l[]。在演算法中,我從此列表中選擇前 n 個整數。我如何編寫乳膠程式碼來執行此操作。
我可以使用 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}