PAT乙级1039. 到底买不买（20）

1039. 到底买不买（20）

小红想买些珠子做一串自己喜欢的珠串。卖珠子的摊主有很多串五颜六色的珠串，但是不肯把任何一串拆散了卖。于是小红要你帮忙判断一下，某串珠子里是否包含了全部自己想要的珠子？如果是，那么告诉她有多少多余的珠子；如果不是，那么告诉她缺了多少珠子。

为方便起见，我们用[0-9]、[a-z]、[A-Z]范围内的字符来表示颜色。例如在图1中，第3串是小红想做的珠串；那么第1串可以买，因为包含了全部她想要的珠子，还多了8颗不需要的珠子；第2串不能买，因为没有黑色珠子，并且少了一颗红色的珠子。

图 1
输入格式：

每个输入包含1个测试用例。每个测试用例分别在2行中先后给出摊主的珠串和小红想做的珠串，两串都不超过1000个珠子。

输出格式：

如果可以买，则在一行中输出“Yes”以及有多少多余的珠子；如果不可以买，则在一行中输出“No”以及缺了多少珠子。其间以1个空格分隔。

输入样例1：
ppRYYGrrYBR2258
YrR8RrY
输出样例1：
Yes 8
输入样例2：
ppRYYGrrYB225
YrR8RrY
输出样例2：
No 2

代码

#include<iostream>
using namespace std;
int main() {
    //摊主珠串，小红want珠串
    char num[1001],wantnum[1001];
    //保存摊主各珠子数量，保存小红want各珠子数量
    int word[128] = {0},wantword[128] = {0};
    cin>>num;
    cin>>wantnum;
    int count1 = 0,count2 = 0,flag = 0; //flag==1代表不能买，0为能买
    //计算摊主各珠子数量
    for (int i = 0; num[i] != '\0'; ++i) {
        word[num[i]]++;
    }
    //计算小红want各珠子数量
    for (int i = 0; wantnum[i] != '\0'; ++i) {
        wantword[wantnum[i]]++;
    }
    //统计多/少的数量
    for (int i = 0; i < 128; ++i) {
        if ( flag == 0 && word[i] > wantword[i]) {
            count1 += word[i] - wantword[i];
        } else if (word[i] < wantword[i]) {
            flag = 1;
            count2 += wantword[i] - word[i];
        }
    }
    if (flag == 1) cout<<"No "<<count2;
    else cout<<"Yes "<<count1;
    return 0;

}