重庆市第10届大学生程序设计大赛 A. 狂小P猜密码 题解

原题链接

题意很好理解，这里不再阐述。

值得注意的是 $d_i$ 为1时，其实我们不需要知道 $i$ 是多少就能求出结果，所以程序中我用 cnt 来记录1的个数。
通过题意可以得出以下结论：
cnt > 4 时，结果就为0，因为这个密码最多有4位，超过4为就不满足至少 $d_i=1$ 的条件
cnt == 4 时，结果显然是个非常简单的排列问题，即 4 * 3 * 2 * 1 24种
cnt == 3 时，就有 3 * 2 * 1 * 3 * 2 * 1 36种
比如有1 5 7需要放进4位密码锁中
我对1 5 7 进行全排列 共有3 * 2 * 1种，然后对这三位数中进行插缝， 有五个缝其中只能四个缝可以合法的插入，剩下的一位就是3 * 2 * 1种，所以为36种
cnt == 2 时，这个不会算排列数。。。所以就打了一个表，数一下 14 种。。。
1 1 1 5
1 1 5 1
1 5 1 1
5 1 1 1

5 5 5 1
5 5 1 5
5 1 5 5
1 5 5 5

1 1 5 5
1 5 1 5
5 1 1 5
5 5 1 1
5 1 5 1
1 5 5 1
cnt == 1时，只有1种

代码

#include <iostream>
using namespace std;

int cnt, t;
int ans[10] = { 0,1,14,36,24 };	// 通过以上分析直接打表

int main()
{
	for (int i = 0; i < 10; i++)
	{
		cin >> t;
		cnt += t;
	}

	cout << ans[cnt] << endl;
	return 0;
}