用Java解决牛客网题目NC586车站建造问题

题目描述

有10^8个村庄排在一条公路上，依次编号为0~10^8-1，相邻村庄距离为1，其中有n个村庄居住着牛牛，居住着牛牛的村庄从小到大依次为a0~an-1，其中保证a0=0.
现在需要建设车站，有两个要求必须被满足：
1、每个有牛牛居住的村庄必须修建车站。
2、相邻车站的距离必须为1或为某个质数。
现给出n和a数组，求需要建设车站的最小数量。

 

思路:完成这倒题目前先做牛客网题目NC527最少素数拆分会使这倒题目简单很多NC527链接：https://www.nowcoder.com/questionTerminal/07d6df2014184decb329de777ba7ff51

 

完成NC527后再来完成这个题目，

放代码：

import java.util.*;


public class Solution {
    /**
     * 
     * @param n int整型 
     * @param a int整型一维数组 
     * @return int整型
     */
   public  int work(int n, int[] a) {
		// write code here
			int sum = n;

		for (int i = 0; i < sum - 1; i++) {

			int temp = a[i + 1] - a[i];// 相邻的两个有牛牛居住的村庄之间的距离
			if (temp <= 3)// 1,2,3都是质数 也就是说a[i]+temp(质数)=a[i+1]所以不需要在这两个村庄之间建车站
				continue;

			else if (zhishu(temp) == true)// 同上
				continue;

			else if (temp % 2 == 0)// 表示两村庄之间的距离是偶数，任何一个偶数都=两个质数之和所以在两村庄之间建立一个车站
				n++;

			else if (zhishu(temp - 2) == true)// 表示temp-2是一个质数，temp=质数+2(质数)，所以需要 见一个车站
				n++;

			else// temp不是偶数，temp-2不是质数，所以temp-2-1是偶数 所以temp=temp-2-1+2+1 所以这两个村庄之间需要建立两个车站
				n += 2;

		}

		return n;

	}

	public static boolean zhishu(int b) {//判断传输的值是不是质数

		boolean c = true;

		for (int j = 2; j < b; j++) {

			if (b % j == 0) {
				c = false;
				break;
			}
		}

		return c;

	}
}