二维数组计算对角个数算法

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fhm6100411074/article/details/79949434

有一个任意二位数组，且元素为0或1， 求该二位数组的对角个数？

对角的定义：四个角都是1的四边形即为一个对角。

/**
 * @desc 计算对角的个数
 *
 */
public class Test {
	public static void main(String[] args) {
		int m = 5;
		int n = 6;
		int a[][] = initTable(m, n);
		print(a, m, n);
		int count = count(a, m, n);
		System.out.println("对角个数为:" + count);
	}

	/**
	 * @param m
	 * @param n
	 * @return
	 * @desc 初始化一个m行n列的数组
	 */
	public static int[][] initTable(int m, int n) {
		int[][] a = new int[m][n];
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				a[i][j] = Math.random() > 0.5 ? 1 : 0; // 设置值为0 或1
			}
		}
		return a;
	}

	/**
	 * @param a
	 * @param m
	 * @param n
	 * @desc 打印这个数组
	 */
	public static void print(int a[][], int m, int n) {
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				System.out.print(a[i][j] + " ");
			}
			System.out.println();
		}
	}

	/**
	 * @param a
	 * @param m
	 * @param n
	 * @return 对角的个数
	 * @desc 计算数组对角个数
	 */
	public static int count(int a[][], int m, int n) {
		int count = 0;
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				if (a[i][j] == 1) {//固定左顶点
					for (int p = i + 1; p < m; p++) {
						for (int q = j + 1; q < n; q++) {
							if (a[p][q] == 1 && a[p][j] == 1 && a[i][q] == 1) {
								count++;
							}
						}
					}
				}
			}
		}
		return count;
	}
}