CF721B Passwords 题解

Content

有一天，小 V 突然忘记了他在 Codehorses 的网站上的密码。但是他有所有网站上的 \(n\) 个密码 \(\{s_i\}_{i=1}^n\)，所以他开始一个一个试。他会先从长度最小的密码开始试，如果有多个长度相等的密码就等概率随机选一个，并且他不会再去试已经试过的密码。输入密码需要 \(1\) 秒，如果连续输错 \(k\) 次密码就需要再等 \(5\) 秒才能重新开始试。已知他的密码是一个字符串 \(t\)。求最好情况和最坏情况下他试密码需要的时间。

数据范围：\(1\leqslant n,k\leqslant 100,0\leqslant |s_i|,|t|\leqslant 100\)。

Solution

我们可以开个桶存储一下所有的 \(|s_i|\)，然后利用桶统计出所有长度 \(<|t|\) 的字符串个数，加进答案里面去，设此时答案为 \(ans\)，那么这时，最好情况的话就只需要再试一次密码，最坏的情况就需要试所有 \(n\) 个密码中长度为 \(|t|\) 的密码。注意需要考虑连续输错需要加时的情况。

Code

int n, k, a[107];
string s[107], t;

int main() {
	scanf("%d%d", &n, &k);
	for(int i = 1; i <= n; ++i) {
		cin >> s[i];
		int len = s[i].size();
		a[len]++;
	}
	cin >> t; int lent = t.size(), ans = 0;
	for(int i = 1; i < lent; ++i) ans += a[i];
    printf("%d %d", (ans + 1) + (ans + 1) / k * 5 - (!((ans + 1) % k) ? 5 : 0), (ans + a[lent]) + (ans + a[lent]) / k * 5 - (!((ans + a[lent]) % k) ? 5 : 0));
	return 0;
}