MiniZinc 数组中字符串值的索引

问题

给定一个 MiniZinc 字符串数组：

int: numStats;
set of int: Stats = 1..numStats;
array[Stats] of string: statNames;

...使用从 MiniZinc 数据文件加载的数据：

numStats = 3;
statNames = ["HEALTH", "ARMOR", "MANA"];

如何查找数组中特定字符串的索引？例如，ARMOR 位于位置 2。

上下文

我需要根据其统计数据的一些限制找到最佳的项目选择。该信息存储在声明如下的二维数组中：

int: numItems;
set of int: Items = 1..numItems;
array[Items, Stats] of float: itemStats;

因此，为了对通过所选项目获得的 ARMOR 的最小数量编写约束，我需要知道 ARMOR 在内部数组中的索引为 2。

由于数据文件是由外部程序生成的，并且统计数据的数量和顺序是动态的，因此我无法对约束中的索引进行硬编码。

一种解决方案（在我的情况下不起作用）

The 迷你锌教程使用一个有趣的技巧来实现类似的目标：

set of int: Colors = 1..3;
int: red = 1;
int: yellow = 2;
int: blue = 3;
array[Colors] of string: name = ["red", "yellow", "blue"];

var Colors: x;
constraint x != red;
output [ name[fix(x)] ];

不幸的是，由于 MiniZinc 数据文件中不允许变量声明，因此这个技巧在我的情况下不起作用。

你可以自己写自定义功能获取字符串数组中字符串的索引：

function int: getIndexOfString(string: str, 
                               array[int] of string: string_array) = 
   sum(  [ if str = string_array[i] 
              then i
           else 0 endif  
          | i in index_set(string_array) ]
   );

在此函数中，我创建一个整数数组，其中位置处的整数i任一等于索引str if string_array[i]=str and 0否则。例如，对于您的示例字符串数组["HEALTH", "ARMOR", "MANA"]和strARMOR结果 int 数组将是[0,2,0].

这就是为什么我可以简单地对 int 数组求和来获取字符串的索引。如果该字符串不存在，则返回值为0，这很好，因为 MiniZinc 中的索引默认从 1 开始。

以下是您如何为第一个示例调用上面的函数：

int: numStats;
set of int: Stats = 1..numStats;
array[Stats] of string: statNames;

numStats = 3;
statNames = ["HEALTH", "ARMOR", "MANA"];

var int: indexOfArmor;

constraint 
   indexOfArmor = getIndexOfString("ARMOR",statNames);  

solve satisfy;  

但请注意，上述功能是有限的并且存在一些缺陷。首先，如果该字符串在数组中多次出现，那么您将收到一个无效索引（所有索引的总和，其中str发生了）。另外，如果您为字符串数组设置了自己的索引（例如(2..6)），那么您将需要调整该功能。