[C++竞赛]：题解——利用枚举法解相关问题

由于核桃oj（宇宙免责声明：不是广）为了不让我写题解，把我封禁了，本人手实在是忍不住，所以，我来这写了：）

今日解题法：枚举

枚举法是一种通过遍历所有可能的情况来寻找问题解决方案的算法策略。它简单直接，适用于解空间有限且计算量可控的问题。

例一：百钱百鸡

题目回顾

B3836 [GESP202303 二级] 百鸡问题

出自洛谷B3836

题目描述

“百鸡问题”是出自我国古代《张丘建算经》的著名数学问题。大意为：

“每只公鸡 $5$ 元，每只母鸡 $3$ 元，每 $3$ 只小鸡 $1$ 元；现在有 $100$ 元，买了 $100$ 只鸡，共有多少种方案？”

小明很喜欢这个故事，他决定对这个问题进行扩展，并使用编程解决：如果每只公鸡 $x$ 元，每只母鸡 $y$ 元，每 $z$ 只小鸡 $1$ 元；现在有 $n$ 元，买了 $m$ 只鸡，共有多少种方案？

输入格式

输入一行，包含五个整数，分别为问题描述中的 $x$，$y$，$z$，$n$，$m$。约定 $1\le x,y,z\le 10$，$1\le n,m\le 1000$。

输出格式

输出一行，包含一个整数 $C$，表示有 $C$ 种方案。

输入输出样例 #1

输入 #1

5 3 3 100 100

输出 #1

4

输入输出样例 #2

输入 #2

1 1 1 100 100

输出 #2

5151

题解

方法思路

为了解决这个扩展的“百鸡问题”，我们需要找到所有可能的公鸡、母鸡和小鸡的组合，使得它们的总价格等于给定的金额 ( n )，并且总数量等于给定的鸡的数量 ( m )。具体步骤如下：

1. 枚举公鸡数量：遍历可能的公鸡数量 ( a )，从 0 到 ( \min(n / x, m) )，确保公鸡的总价格不超过剩余金额且总数不超过 ( m )。
2. 枚举母鸡数量：对于每个公鸡数量 ( a )，遍历可能的母鸡数量 ( b )，从 0 到 ( \min(\text{剩余金额} / y, m - a) )，确保母鸡的总价格不超过剩余金额且总数不超过剩余鸡的数量。
3. 计算小鸡数量：对于每个公鸡和母鸡的组合，计算剩余金额能购买的小鸡数量。小鸡的数量为 ( \text{剩余金额} \times z )，同时需要满足总数量 ( a + b + \text{小鸡数量} = m )。

解决代码

#include <iostream>
#include <algorithm>

using namespace std;

int main() {
    int x, y, z, n, m;
    cin >> x >> y >> z >> n >> m;
    
    int ans = 0;
    // 枚举公鸡数量a
    for (int a = 0; a <= min(n / x, m); ++a) {
        int remaining_money = n - a * x;
        if (remaining_money < 0) break;
        
        // 枚举母鸡数量b
        int max_b = min(remaining_money / y, m - a);
        for (int b = 0; b <= max_b; ++b) {
            int remaining = remaining_money - b * y;
            if (remaining < 0) continue;
            
            int chicks = remaining * z;  // 小鸡数量
            if (a + b + chicks == m) {
                ans++;
            }
        }
    }
    
    cout << ans << endl;
    return 0;
}

代码解释

1. 输入处理：读取五个整数 ( x )、( y )、( z )、( n )、( m )，分别表示公鸡价格、母鸡价格、小鸡的每 ( z ) 只价格、总金额和总数量。
2. 初始化答案变量：ans 初始化为 0，用于统计有效方案的数量。
3. 枚举公鸡数量：外层循环遍历可能的公鸡数量 ( a )，从 0 到 ( \min(n / x, m) )。确保公鸡的总价格不超过剩余金额且总数不超过 ( m )。
4. 计算剩余金额：对于每个公鸡数量 ( a )，计算剩余金额 remaining_money。如果剩余金额为负，跳出循环。
5. 枚举母鸡数量：内层循环遍历可能的母鸡数量 ( b )，从 0 到 ( \min(\text{remaining_money} / y, m - a) )。确保母鸡的总价格不超过剩余金额且总数不超过剩余鸡的数量。
6. 计算小鸡数量：对于每个公鸡和母鸡的组合，计算剩余金额能购买的小鸡数量 chicks = remaining * z。检查总数量是否等于 ( m )，如果是，则增加方案数量 ans。
7. 输出结果：打印有效方案的数量 ans。

这种方法通过双重循环枚举所有可能的公鸡和母鸡组合，并验证小鸡数量是否满足条件，确保在给定的约束下高效地找到所有有效方案。

例二：平方之和

题目回顾

B4002 [GESP202406 二级] 平方之和

题目描述

小杨有 $n$ 个正整数 $a_1,a_2,\dots ,a_n$，他想知道对于所有的 $i(1\le i\le n)$，是否存在两个正整数 $x$ 和 $y$ 满足 $x\times x+y\times y=a_i$。

输入格式

第一行包含一个正整数 $n$，代表正整数数量。
之后 $n$ 行，每行包含一个正整数，代表 $a_i$。

输出格式

对于每个正整数 $a_i$，如果存在两个正整数 $x$ 和 $y$ 满足 $x\times x+y\times y=a_i$，输出 Yes，否则输出 No。

输入输出样例 #1

输入 #1

2
5
4

输出 #1

Yes
No

说明/提示

对于第一个正整数，存在 $1\times 1+2\times 2=5$，因此答案为 Yes。

对于全部数据，保证有 $1\le n\le 10,1\le a_i\le 10^6$。

题解

题解：P5749 平方之和

题目分析

题目要求我们判断给定的每个正整数是否可以表示为两个正整数的平方和。即对于每个数a_i，是否存在正整数x和y，使得x² + y² = a_i。

解题思路

1. 遍历可能的x值：对于每个给定的数num，我们可以从x=1开始尝试，直到x²超过num。
2. 计算剩余部分：对于每个x，计算y² = num - x²。
3. 检查y是否为整数：计算y的平方根，并检查其平方是否等于y²，同时确保y为正整数。

代码实现

#include <bits/stdc++.h>
using namespace std;

bool is_sum_of_squares(int num) {
    // 检查是否存在两个正整数x和y，使得x² + y² = num
    for (int x = 1; x * x < num; ++x) {
        int y_squared = num - x * x;
        int y = sqrt(y_squared);
        if (y * y == y_squared && y > 0) {
            return true;
        }
    }
    return false;
}

int main() {
    int n;
    cin >> n;
    
    for (int i = 0; i < n; ++i) {
        int a;
        cin >> a;
        
        if (is_sum_of_squares(a)) {
            cout << "Yes" << endl;
        } else {
            cout << "No" << endl;
        }
    }
    
    return 0;
}

代码解释

• is_sum_of_squares函数：该函数用于判断一个数是否可以表示为两个正整数的平方和。通过遍历x的可能值，计算对应的y²，并检查y是否为整数。
• 主函数：读取输入的n个数，对每个数调用is_sum_of_squares函数进行判断，并输出结果。

例三：揭竿而起

题目回顾

题解

题解：P5754 揭竿而起

题目分析

题目要求从9支队伍中选出7支，使得这7支队伍的总人数恰好为100人。每支队伍的人数各不相同且不超过100。需要输出这7支队伍的编号（按输入顺序编号），按编号从小到大排列。

解题思路

1. 组合枚举：由于需要从9支队伍中选出7支，可以使用位运算枚举所有可能的组合（共C(9,7)=36种）。
2. 人数检查：对于每种组合，计算7支队伍的总人数，检查是否等于100。
3. 输出结果：找到符合条件的组合后，将队伍编号排序并输出。

代码实现

#include <bits/stdc++.h>
using namespace std;

const int TOTAL_TEAMS = 9;
const int SELECTED_TEAMS = 7;
const int TARGET_PEOPLE = 100;

int main() {
    vector<int> team_sizes(TOTAL_TEAMS);
    // 读取每支队伍的人数
    for (int i = 0; i < TOTAL_TEAMS; ++i) {
        cin >> team_sizes[i];
    }

    // 生成所有可能的组合，用位运算枚举
    for (int mask = 0; mask < (1 << TOTAL_TEAMS); ++mask) {
        // 检查是否恰好选了7支队伍
        if (__builtin_popcount(mask) == SELECTED_TEAMS) {
            int total_people = 0;
            vector<int> selected_teams;
            for (int i = 0; i < TOTAL_TEAMS; ++i) {
                if (mask & (1 << i)) {
                    total_people += team_sizes[i];
                    selected_teams.push_back(i + 1);  // 队伍编号从1开始
                }
            }
            // 检查总人数是否为100
            if (total_people == TARGET_PEOPLE) {
                sort(selected_teams.begin(), selected_teams.end());
                for (int team_id : selected_teams) {
                    cout << team_id << endl;
                }
                break;  // 题目保证唯一解，找到后直接退出
            }
        }
    }

    return 0;
}

代码解释

• 输入处理：读取9支队伍的人数，存储在team_sizes数组中。
• 组合枚举：通过位运算枚举所有可能的7支队伍组合（mask的二进制表示中恰好有7个1）。
• 人数检查：对于每种组合，计算总人数，如果等于100，则记录队伍编号。
• 输出结果：将符合条件的队伍编号排序后输出。