Rake It In ( DFS)

计蒜客Rake It In

由于k<=3,t<=200,可以想到,在最坏的情况下,暴力计算出每一层的最优解,一共也只有6层,时间最多为200*(9^6)。暴力递归(回溯)出每一层的最优解,进而求取最终答案。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;

int a[5][5];
const int n = 4;
int k;

void cc(int i,int j)
{
    int tmp=a[i][j];
    a[i][j]=a[i][j+1];
    a[i][j+1]=a[i+1][j+1];
    a[i+1][j+1]=a[i+1][j];
    a[i+1][j]=tmp;
}

void ee(int i,int j)
{
    int tmp=a[i][j];
    a[i][j]=a[i+1][j];
    a[i+1][j]=a[i+1][j+1];
    a[i+1][j+1]=a[i][j+1];
    a[i][j+1]=tmp;
}

int dd(int x,int y)
{
    return a[x][y] + a[x + 1][y] + a[x][y + 1] + a[x + 1][y + 1];
}

int DFS(int num)
{
    int ans = 0;
    if (num == 2 * k)
    {
        int mini = inf;
        for (int i = 1; i < n; i++)
        {
            for (int j = 1; j < n;j++)
            {
                mini = min(mini, dd(i, j));
            }
        }
        return mini;
    }
    else if(num%2==1)
    {
        int mx = 0;
        for (int i = 1; i < n; i++)
        {
            for (int j = 1; j < n;j++)
            {
                cc(i, j);
                int sum = dd(i, j) + DFS(num + 1);
                mx = max(mx, sum);
                ee(i, j);
            }
        }
        return mx;
    }
    else if(num%2==0)
    {
        int mini = inf;
        for (int i = 1; i < n; i++)
        {
            for (int j = 1; j < n;j++)
            {
                cc(i, j);
                int sum = dd(i, j) + DFS(num + 1);
                mini = min(mini, sum);
                ee(i, j);
            }
        }
        return mini;
    }
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &k);
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n;j++)
                scanf("%d", &a[i][j]);
        printf("%d\n", DFS(1));
    }
    return 0;
}

 

你可能感兴趣的:(题目)