PAT乙级 1039 到底买不买
题目
小红想买些珠子做一串自己喜欢的珠串。卖珠子的摊主有很多串五颜六色的珠串,但是不肯把任何一串拆散了卖。于是小红要你帮忙判断一下,某串珠子里是否包含了全部自己想要的珠子?如果是,那么告诉她有多少多余的珠子;如果不是,那么告诉她缺了多少珠子。
为方便起见,我们用[0-9]、[a-z]、[A-Z]范围内的字符来表示颜色。例如在图1中,第3串是小红想做的珠串;那么第1串可以买,因为包含了全部她想要的珠子,还多了8颗不需要的珠子;第2串不能买,因为没有黑色珠子,并且少了一颗红色的珠子。
输入格式:
每个输入包含 1 个测试用例。每个测试用例分别在 2 行中先后给出摊主的珠串和小红想做的珠串,两串都不超过 1000 个珠子。
输出格式:
如果可以买,则在一行中输出 Yes 以及有多少多余的珠子;如果不可以买,则在一行中输出 No 以及缺了多少珠子。其间以 1 个空格分隔。
输入样例 1:
ppRYYGrrYBR2258
YrR8RrY
输出样例 1:
Yes 8
输入样例 2:
ppRYYGrrYB225
YrR8RrY
输出样例 2:
No 2
题解
#include <bits/stdc++.h>
int main() {
std::string shop_beads, want_beads;
std::cin >> shop_beads >> want_beads;
if(want_beads.size() > shop_beads.size()) {
std::cout << "No " << want_beads.size() - shop_beads.size() << std::endl;
}
std::unordered_map<char, int> m_1, m_2;
for(const auto& c : shop_beads) {
if(m_1.find(c) == m_1.end()) {
m_1[c] = 1;
}else {
m_1[c]++;
}
}
for(int i = 0; i < want_beads.size(); i++) {
if(m_2.find(want_beads[i]) == m_2.end()) {
m_2[want_beads[i]] = 1;
}else {
m_2[want_beads[i]]++;
}
}
bool is_miss = false;
int miss = 0, extra = 0;
for(int i = 0; i < want_beads.size(); i++) {
char c = want_beads[i];
if(m_1.find(c) == m_1.end()) {//没找到
is_miss = true;
}else {
if(m_1[c] <= 0) is_miss = true;
m_1[c]--;
m_2[c]--;
}
}
if(is_miss) {
for(const auto& pair : m_2) {
if(pair.second != 0) miss += pair.second;
}
for(const auto& pair : m_1) {
if(pair.second < 0) miss += std::abs(pair.second);
}
std::cout << "No " << miss << std::endl;
}else {
for(const auto& pair : m_1) {
if(pair.second != 0) extra += pair.second;
}
std::cout << "Yes " << extra << std::endl;
}
return 0;
}