POJ2159密码问题

题目链接：http://poj.org/problem?id=2159

思路：只要用两个数组分别存下两个字符串中各个字母的个数，排序一下，比较字母个数是不是都相等就可以了，如果不是全相等，则说明不能从原文信息得到加密信息。如果全相等，则一定有方法从原文信息得到加密信息。

代码如下：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;
int main()
{
    int a[30],b[30];
    char str1[200],str2[200];
    while(scanf("%s%s",str1,str2) != EOF)
    {
        int i = 0;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        int len1 = strlen(str1);
        int len2 = strlen(str2);
        for(i=0; i<len1; i++)
            a[str1[i]-'A']++;
        for(i=0; i<len2; i++)
            b[str2[i]-'A']++;
        sort(a,a+26);
        sort(b,b+26);
        int flag = 1;
        for(i=0; i<26; i++)
        {
            if(a[i] != b[i])
            {
                flag = 0;
                break;
            }
        }
        if(flag)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}