USACO - Chapter1 Section 1.3 - Prime Cryptarithm

Prime Cryptarithm

题目描述

下面是一个乘法竖式，如果用我们给定的那n个数字来取代*，可以使式子成立的话，我们就叫这个式子牛式。

      ***
x     **

---

     ***
    ***

---

    ****

（请复制到记事本）

数字只能取代*，当然第一位不能为0,况且给定的数字里不包括0。

注意一下在美国的学校中教的“部分乘积”，第一部分乘积是第二个数的个位和第一个数的积，第二部分乘积是第二个数的十位和第一个数的乘积.

写一个程序找出所有的牛式。

输入输出格式

输入格式：
Line 1:数字的个数n。

Line 2:N个用空格分开的数字（每个数字都属于{1,2,3,4,5,6,7,8,9}）。

输出格式：
共一行，一个数字。表示牛式的总数。

输入输出样例

输入样例#1：
5
2 3 4 6 8
输出样例#1：
1

完全枚举，代码丑出天际了。。

Code

#include<iostream>
#include<cstdio>
#include<set>
using namespace std;
int main()
{
    set<int> s;
    int n;
    cin >> n;
    int a[n];
    for (int i=0;i<n;i++)
    {
        cin >> a[i];
        s.insert(a[i]);
    }
    int ans = 0;
    for (int i=0;i<n;i++)
    {
        for (int j=0;j<n;j++)
        {
            for (int k=0;k<n;k++)
            {
                for (int l=0;l<n;l++)
                {
                    for (int m=0;m<n;m++)
                    {
                        int top = a[i]*100+a[j]*10+a[k];
                        int gewei = a[m]*top;
                        int shiwei = a[l]*top;
                        int chengji = gewei+shiwei*10;
                        int aaa = gewei/100;
                        int bbb = (gewei/10)%10;
                        int ccc = gewei%10;
                        int ddd = shiwei/100;
                        int eee = (shiwei/10)%10;
                        int fff = shiwei%10;
                        int ggg = chengji/1000;
                        int hhh = (chengji/10) %10;
                        int iii = (chengji/100) %10;
                        int jjj = chengji%10;
                        if (!s.count(aaa)||!s.count(bbb)||!s.count(ccc)||!s.count(ddd)||!s.count(eee)||!s.count(fff)||!s.count(ggg)||!s.count(hhh)||!s.count(iii)||!s.count(jjj))
                            continue;
                        else 
                            ans++;
                    }
                }
            }
        }
    }
    cout << ans;
    return 0;
}