📜  金矿问题

📅  最后修改于: 2021-09-22 10:36:34             🧑  作者: Mango

给定一个 n*m 尺寸的金矿。该矿场的每个字段都包含一个正整数,表示以吨为单位的黄金量。最初矿工在第一列,但可以在任何行。他只能从给定的单元格移动(右->,右上/,右下\),矿工可以向右或向右斜向上或向右斜向下移动到单元格。找出他可以收集的最大黄金量。
例子:

Input : mat[][] = {{1, 3, 3},
                   {2, 1, 4},
                  {0, 6, 4}};
Output : 12 
{(1,0)->(2,1)->(2,2)}

Input: mat[][] = { {1, 3, 1, 5},
                   {2, 2, 4, 1},
                   {5, 0, 2, 3},
                   {0, 6, 1, 2}};
Output : 16
(2,0) -> (1,1) -> (1,2) -> (0,3) OR
(2,0) -> (3,1) -> (2,2) -> (2,3)

Input : mat[][] = {{10, 33, 13, 15},
                  {22, 21, 04, 1},
                  {5, 0, 2, 3},
                  {0, 6, 14, 2}};
Output : 83

来源 Flipkart 采访

创建与给定矩阵 mat[][] 相同的二维矩阵 goldTable[][])。如果我们仔细观察这个问题,我们可以注意到以下内容。

  1. 黄金的数量是正数,因此我们希望在给定的约束下覆盖最大值的最大单元格。
  2. 每一步,我们都向右侧移动一步。所以我们总是在最后一列结束。如果我们在最后一列,那么我们无法向右移动

如果我们在第一行或最后一列,那么我们无法向右移动,所以只分配 0 否则将 goldTable[row-1][col+1] 的值分配给 right_up。如果我们在最后一行或最后一列,那么我们无法向右移动,所以只分配 0 否则将 goldTable[row+1][col+1] 的值分配到右边。
现在找到 right、right_up 和 right_down 的最大值,然后将其与 mat[row][col] 相加。最后,找到所有行和第一列的最大值并返回它。

C++
// C++ program to solve Gold Mine problem
#include
using namespace std;
 
const int MAX = 100;
 
// Returns maximum amount of gold that can be collected
// when journey started from first column and moves
// allowed are right, right-up and right-down
int getMaxGold(int gold[][MAX], int m, int n)
{
    // Create a table for storing intermediate results
    // and initialize all cells to 0. The first row of
    // goldMineTable gives the maximum gold that the miner
    // can collect when starts that row
    int goldTable[m][n];
    memset(goldTable, 0, sizeof(goldTable));
 
    for (int col=n-1; col>=0; col--)
    {
        for (int row=0; row)
            int right = (col==n-1)? 0: goldTable[row][col+1];
 
            // Gold collected on going to the cell to right up (/)
            int right_up = (row==0 || col==n-1)? 0:
                            goldTable[row-1][col+1];
 
            // Gold collected on going to the cell to right down (\)
            int right_down = (row==m-1 || col==n-1)? 0:
                             goldTable[row+1][col+1];
 
            // Max gold collected from taking either of the
            // above 3 paths
            goldTable[row][col] = gold[row][col] +
                              max(right, max(right_up, right_down));
                                                     
        }
    }
 
    // The max amount of gold collected will be the max
    // value in first column of all rows
    int res = goldTable[0][0];
    for (int i=1; i


Java
// Java program to solve Gold Mine problem
import java.util.Arrays;
 
class GFG {
     
    static final int MAX = 100;
     
    // Returns maximum amount of gold that
    // can be collected when journey started
    // from first column and moves allowed
    // are right, right-up and right-down
    static int getMaxGold(int gold[][],
                              int m, int n)
    {
         
        // Create a table for storing
        // intermediate results and initialize
        // all cells to 0. The first row of
        // goldMineTable gives the maximum
        // gold that the miner can collect
        // when starts that row
        int goldTable[][] = new int[m][n];
         
        for(int[] rows:goldTable)
            Arrays.fill(rows, 0);
     
        for (int col = n-1; col >= 0; col--)
        {
            for (int row = 0; row < m; row++)
            {
                 
                // Gold collected on going to
                // the cell on the right(->)
                int right = (col == n-1) ? 0
                        : goldTable[row][col+1];
     
                // Gold collected on going to
                // the cell to right up (/)
                int right_up = (row == 0 ||
                               col == n-1) ? 0 :
                        goldTable[row-1][col+1];
     
                // Gold collected on going to
                // the cell to right down (\)
                int right_down = (row == m-1
                            || col == n-1) ? 0 :
                          goldTable[row+1][col+1];
     
                // Max gold collected from taking
                // either of the above 3 paths
                goldTable[row][col] = gold[row][col]
                 + Math.max(right, Math.max(right_up,
                                       right_down));
                                                         
            }
        }
     
        // The max amount of gold collected will be
        // the max value in first column of all rows
        int res = goldTable[0][0];
         
        for (int i = 1; i < m; i++)
            res = Math.max(res, goldTable[i][0]);
             
        return res;
    }
     
    //driver code
    public static void main(String arg[])
    {
        int gold[][]= { {1, 3, 1, 5},
                        {2, 2, 4, 1},
                        {5, 0, 2, 3},
                        {0, 6, 1, 2} };
                         
        int m = 4, n = 4;
         
        System.out.print(getMaxGold(gold, m, n));
    }
}
 
// This code is contributed by Anant Agarwal.


Python3
# Python program to solve
# Gold Mine problem
 
MAX = 100
 
# Returns maximum amount of
# gold that can be collected
# when journey started from
# first column and moves
# allowed are right, right-up
# and right-down
def getMaxGold(gold, m, n):
 
    # Create a table for storing
    # intermediate results
    # and initialize all cells to 0.
    # The first row of
    # goldMineTable gives the
    # maximum gold that the miner
    # can collect when starts that row
    goldTable = [[0 for i in range(n)]
                        for j in range(m)]
 
    for col in range(n-1, -1, -1):
        for row in range(m):
 
            # Gold collected on going to
            # the cell on the rigth(->)
            if (col == n-1):
                right = 0
            else:
                right = goldTable[row][col+1]
 
            # Gold collected on going to
            # the cell to right up (/)
            if (row == 0 or col == n-1):
                right_up = 0
            else:
                right_up = goldTable[row-1][col+1]
 
            # Gold collected on going to
            # the cell to right down (\)
            if (row == m-1 or col == n-1):
                right_down = 0
            else:
                right_down = goldTable[row+1][col+1]
 
            # Max gold collected from taking
            # either of the above 3 paths
            goldTable[row][col] = gold[row][col] + max(right, right_up, right_down)
                                                            
    # The max amount of gold
    # collected will be the max
    # value in first column of all rows
    res = goldTable[0][0]
    for i in range(1, m):
        res = max(res, goldTable[i][0])
 
    return res
     
# Driver code
gold = [[1, 3, 1, 5],
    [2, 2, 4, 1],
    [5, 0, 2, 3],
    [0, 6, 1, 2]]
 
m = 4
n = 4
 
print(getMaxGold(gold, m, n))
 
# This code is contributed
# by Soumen Ghosh.


C#
// C# program to solve Gold Mine problem
using System;
 
class GFG
{
    static int MAX = 100;
 
    // Returns maximum amount of gold that
    // can be collected when journey started
    // from first column and moves allowed are
    // right, right-up and right-down
    static int getMaxGold(int[,] gold,
                            int m, int n)
    {
         
        // Create a table for storing intermediate
        // results and initialize all cells to 0.
        // The first row of goldMineTable gives
        // the maximum gold that the miner
        // can collect when starts that row
        int[,] goldTable = new int[m, n];
         
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                goldTable[i, j] = 0;
     
        for (int col = n - 1; col >= 0; col--)
        {
            for (int row = 0; row < m; row++)
            {
                // Gold collected on going to
                // the cell on the right(->)
                int right = (col == n - 1) ? 0 :
                            goldTable[row, col + 1];
     
                // Gold collected on going to
                // the cell to right up (/)
                int right_up = (row == 0 || col == n - 1)
                            ? 0 : goldTable[row-1,col+1];
     
                // Gold collected on going
                // to the cell to right down (\)
                int right_down = (row == m - 1 || col == n - 1)
                                ? 0 : goldTable[row + 1, col + 1];
     
                // Max gold collected from taking
                // either of the above 3 paths
                goldTable[row, col] = gold[row, col] +
                                Math.Max(right, Math.Max(right_up,
                                                    right_down));
            }
        }
     
        // The max amount of gold collected will be the max
        // value in first column of all rows
        int res = goldTable[0, 0];
        for (int i = 1; i < m; i++)
            res = Math.Max(res, goldTable[i, 0]);
        return res;
    }
     
    // Driver Code
    static void Main()
    {
        int[,] gold = new int[,]{{1, 3, 1, 5},
                                {2, 2, 4, 1},
                                {5, 0, 2, 3},
                                {0, 6, 1, 2}
                                };
        int m = 4, n = 4;
        Console.Write(getMaxGold(gold, m, n));
    }
}
 
// This code is contributed by DrRoot_


PHP
= 0 ; $col--)
    {
        for ($row = 0 ; $row < $m ; $row++)
        {
 
            // Gold collected on going to
            // the cell on the rigth(->)
            if ($col == $n - 1)
                $right = 0 ;
            else
                $right = $goldTable[$row][$col + 1];
 
            // Gold collected on going to
            // the cell to right up (/)
            if ($row == 0 or $col == $n - 1)
                $right_up = 0 ;
            else
                $right_up = $goldTable[$row - 1][$col + 1];
 
            // Gold collected on going to
            // the cell to right down (\)
            if ($row == $m - 1 or $col == $n - 1)
                $right_down = 0 ;
            else
                $right_down = $goldTable[$row + 1][$col + 1];
 
            // Max gold collected from taking
            // either of the above 3 paths
            $goldTable[$row][$col] = $gold[$row][$col] +
                                 max($right, $right_up,
                                             $right_down);
        }
    }
     
    // The max amount of gold collected will be the
    // max value in first column of all rows
    $res = $goldTable[0][0] ;
    for ($i = 0; $i < $m; $i++)
        $res = max($res, $goldTable[$i][0]);
 
    return $res;
}
     
// Driver code
$gold = array(array(1, 3, 1, 5),
              array(2, 2, 4, 1),
              array(5, 0, 2, 3),
              array(0, 6, 1, 2));
 
$m = 4 ;
$n = 4 ;
 
echo getMaxGold($gold, $m, $n) ;
 
// This code is contributed by Ryuga
?>


Javascript


输出

16

如果您希望与专家一起参加现场课程,请参阅DSA 现场工作专业课程学生竞争性编程现场课程