Java洛谷P1149 火柴棒等式

题目链接:https://www.luogu.com.cn/problem/P1149

代码实例:

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int input = scanner.nextInt();
		int count = 0;
		int c[] = new int[2001];
		c[0] = 6;
		//a存储0-9每个数字所用的火柴数目
		int a[] = new int [] {
				6,2,5,5,4,5,6,3,7,6
			};
		
		for(int i=1;i<=2000;i++) {
            int j=i;
            while(j>=1)//求每个数所用的火柴棒
            {
                    c[i] += a[j%10];
                    j=j/10;
            }
        }
		for (int i = 0; i < 1000; i++) {
			for (int j = 0; j < 1000; j++) {
					if(c[i]+c[j]+c[i+j]+4==input) {
						count++;
				}
			}
		}
		System.out.println(count);
		scanner.close();
	}
}