USACO历年青铜组真题解析 | 2023年1月Leaders

欢迎大家订阅我的专栏：算法题解：C++与Python实现！
本专栏旨在帮助大家从基础到进阶 ，逐步提升编程能力，助力信息学竞赛备战！

专栏特色
1.经典算法练习：根据信息学竞赛大纲，精心挑选经典算法题目，提供清晰的代码实现与详细指导，帮助您夯实算法基础。
2.系统化学习路径：按照算法类别和难度分级，从基础到进阶，循序渐进，帮助您全面提升编程能力与算法思维。

适合人群：

• 准备参加蓝桥杯、GESP、CSP-J、CSP-S等信息学竞赛的学生
• 希望系统学习C++/Python编程的初学者
• 想要提升算法与编程能力的编程爱好者

附上汇总贴：USACO历年青铜组真题解析 | 汇总-CSDN博客

---

【题目描述】

FJ 有 N 头奶牛，每一头奶牛的品种是根西岛 G 或荷斯坦 H 中的一种。

每一头奶牛都有一个名单，第 i 头奶牛的名单上记录了从第 i 头奶牛到第 Ei 头奶牛的所有奶牛。

每一种奶牛都有且仅有一位“领导者”，对于某一头牛 i，如果它能成为“领导者”仅当它满足以下条件的至少一个：

• 其记录的名单上包含它的品种的所有奶牛。
• 其记录的名单上记录了另一品种奶牛的“领导者”。

请求出有多少对奶牛可能成为两种奶牛的领导者，保证存在至少一种。

【输入】

The first line contains N.

The second line contains a string of length N , with the ith character denoting the breed of the i-th cow (G meaning Guernsey and H meaning Holstein). It is guaranteed that there is at least one Guernsey and one Holstein.

The third line contains E1⋯EN.

【输出】

Output the number of possible pairs of leaders.

【输入样例】

4
GHHG
2 4 3 4

【输出样例】

1

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int n, e[200005], firstH=0, lastH=0, firstG=0, lastG=0, ansG[200005]={0}, ansH[200005]={0}, cntG=0, cntH=0;  // ans1记录该头牛是否包含所有牛
int leaderG=0, leaderH=0;
string s;
int main()
{
    cin >> n;  // 输入n
    cin >> s;  // 输入字符串（方便查找）
    s = " " + s;  // 因为起始位置为1，所以在字符串前面加个空格
    for (int i=1; i<=n; i++) {  // 遍历n头奶牛
        cin >> e[i];  // 记录每头奶牛的名单
    }
    firstG = s.find('G');  // 获取第一头G奶牛的位置
    lastG = s.rfind('G');  // 获取最后一头G奶牛的位置
    firstH = s.find('H');  // 获取第一头H奶牛的位置
    lastH = s.rfind('H');  // 获取最后一头H奶牛的位置
    for (int i=1; i<=n; i++) {  // 遍历所有奶牛，找到包含所有同种品种奶牛的位置
        if (s[i]=='G') {  // 找G牛
            if (i<=firstG && e[i]>=lastG) {  // 如果包含所有同种品种的奶牛
                ansG[i]=1;  // 标记为1
                leaderG = i;  // 并且记录位置（G中有且只有一头头牛）
            }
        }
        if (s[i]=='H') {  // 找H牛
            if (i<=firstH && e[i]>=lastH) {  // 如果包含所有同品种的奶牛
                ansH[i]=1;  // 标记为1
                leaderH = i; // 并记录位置（H中也有且只有一头头牛）
            }
        }
    }
    for (int i=1; i<=n; i++) {  // 再次遍历n头奶牛
        if (s[i]=='G') {  // 查找满足条件：名单上记录另一品种奶牛的领导者
            if (i<=leaderH && e[i]>=leaderH) ansG[i]= 1;  // 满足条件的标记为1
        }
        if (s[i]=='H') {  // 同上
            if (i<=leaderG && e[i]>=leaderG) ansH[i]= 1;
        }
    }
    for (int i=1; i<=n; i++) {  // 最后将G牛和H牛种中标记为1的牛进行统计
        if (ansG[i]==1) cntG++;
        if (ansH[i]==1) cntH++;
    }
    cout << cntG*cntH << endl;  // 将两者统计结果相乘，得到方案数
    return 0;
}

【运行结果】

4
GHHG
2 4 3 4
1