PTA 找单词 (15分)（bfs）

7-2 找单词 (15分)

假设有x1个字母A， x2个字母B,… x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,… 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词呢？单词的价值就是组成一个单词的所有字母的价值之和，比如，单词ACM的价值是1+3+14=18，单词HDU的价值是8+4+21=33。(组成的单词与排列顺序无关，比如ACM与CMA认为是同一个单词）。

输入格式:

输入首先是一个整数N，代表测试实例的个数。 然后包括N行数据，每行包括26个<=20的整数x1,x2,…x26.

输出格式:

对于每个测试实例，请输出能找到的总价值<=50的单词数,每个实例的输出占一行。

输入样例:

2
1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
9 2 6 2 10 2 2 5 6 1 0 2 7 0 2 2 7 5 10 6 10 2 10 6 1 9

输出样例:

在这里给出相应的输出。例如：

7
379297

解题

这题比较清晰，直接用bfs搜索，注意每个字母会搜索多次而不是一次，还有从每次都从下一个节点入队，避免重复。

代码

#include <algorithm>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
const int maxn = 27;
int x[maxn];

struct node {
    
    
    int v, k, sum;  //顶点编号, 表示剩余次数 + 1, 当前节点的
};

int bfs() {
    
    
    queue<node> q;
    for (int i = 1; i < 27; i++)
        if (x[i] > 0) q.push({
    
    i, x[i], i});

    int ans = 0;
    while (!q.empty()) {
    
    
        int v = q.front().v;
        int k = q.front().k;
        int sum = q.front().sum;
        q.pop();
        if (sum <= 50)
            ans++;
        else
            continue;

        if (k > 1) q.push({
    
    v, k - 1, sum + v});
        for (int i = v + 1; i < 27; i++)
            if (x[i] > 0) q.push({
    
    i, x[i], sum + i});
    }
    return ans;
}

int main() {
    
    
    int N;
    cin >> N;
    while (N--) {
    
    
        for (int i = 1; i < 27; i++) cin >> x[i];
		
        cout << bfs() << endl;
    }

    system("pause");
    return 0;
}