面試題 16.11. 跳水板

 

 

public static void main(String[] args) {
		
		 int[] divingBoard = divingBoard(1,2,0);
		 for (int i : divingBoard) {
			System.out.println(i);
		}
		 
	}

    //本道題是在遞歸題裏面找到的。但是我找不到遞推公式。 所以呢用了一種直接簡單明瞭的方式
    //大家很容易懂得一種方法

	public  static int[] divingBoard(int shorter, int longer, int k) {

		int[] dp = new int[k+1];
		
        //如果k==0直接返回空數組
		if(k == 0)
		{
			dp = new int[0];
			return dp;
		}
		
        //如果兩塊木板得長度相等得話  那麼只有一種長度
		if(shorter == longer)
		{
			dp = new int[1];
			dp[0] = shorter*k;
			return dp;
		}
		
        //如果只讓使用一個木板得話，那只有兩種長度
		if(k == 1)
		{
			dp[0] = shorter;
			dp[1] = longer;
			return dp;
		}
		//如果k>1並且shorter!=longer的時候，題目要求長度從小到大輸出。那麼前面的元素肯定是用
        //最多的短木板的組成的一個長度值。
		for(int i = 0;i<=k;i++)
		{
			dp[i] = shorter*(k-i)+longer*i;
		}
		
		return dp;
    }