[USACO1.3.4]牛式 Prime Cryptarithm JAVA实现

[USACO1.3.4]牛式 Prime Cryptarithm JAVA实现
7-13 1.3.4 牛式 (70分)

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

       * * * 	x    * * 	--------------
       * * *
     * * *
    --------
    * * * *

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

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

写一个程序找出所有的牛式。 输入格式:

Line 1: 数字的个数. Line 2: N 个用空格分开的数字(每个数字都∈{1,2,3,4,5,6,7,8,9}) . 输出格式:

共一行,一个数字.表示牛式的总数.下面是样例的那个牛式.

2 2 2

• 2 2

• • • • • •  4 4 4 4 4 4 
            

• • • • • • 4 8 8 4

输入样例:

在这里给出一组输入。例如：

5 2 3 4 6 8

输出样例:

在这里给出相应的输出。例如：

1

思路
这题可以直接枚举一个两位数和一个三位数相乘，有两个判断的函数：
一个是判断是否符合两位数的个位乘以三位数是一个三位数，两位数的十位乘以三位数也是一个三位数，第二个结果乘以十加上第一个结果是否是四位数。
第二个判断函数是要判断得到的数字是否是输入进来的数字，这里尤其要注意，不仅仅是枚举出来的两位数和三位数要进行判断，第一个判断函数里面得到的数字也要进行判断。

import java.util.ArrayList;
import java.util.Scanner;

/**
 * 7-13 1.3.4 牛式
 */
public class PrimeCryptarithm {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n =scan.nextInt();
        ArrayList<Integer> number=new ArrayList<Integer>();
        for (int i =0;i<n;i++){
            Integer temp = scan.nextInt();
            number.add(temp);
        }
        int cnt=0;
        for(int i =10;i<100;i++){
            for (int j = 100;j<1000;j++){
                if(cheak(j,i,number)&&isInput(number,i)&&isInput(number,j)){
                    cnt++;
                }
            }
        }
        System.out.println(cnt);
    }

    /**
     * 判断i中的数字是否是输入的数字
     * @param num
     * @param i
     * @return
     */
    public static boolean isInput(ArrayList num,Integer i){
        while (i!=0){
            if(!num.contains(i%10))
                return false;
            i/=10;
        }
        return true;
    }

    public static boolean cheak(int tens,int unit,ArrayList num){
        int a = tens*(unit/10);//两位数的十位乘以三位数
        int b= tens*(unit%10);//两位数的个位乘以三位数
        int number = a*10+b;
        if(a>999||b>999||number>9999)
            return false;
        if(!isInput(num,a))
            return false;
        if(!isInput(num,b))
            return false;
        if(!isInput(num,number))
            return false;
        return true;
    }