倒水问题BFS CS精英挑战营

倒水问题

问题分析
操作分类：本质上是六种操作。
如果采用暴力枚举的方法，O(6^t) （t<=4 可以用搜索算法）
采用动态规划算法。 与记忆化搜索类似。通过搜索算法去优化。 只要A杯水u中和B杯水v中的量是相同的，并且操作数为i，dp[i][u][v]，那么就可以认为是相同的状态。dp[i+1][x][y] 来表示下一次可以到达的水量x,y。 若为零则表示不能到达。若为一，则表示能到达。O(nmt)—>该法可以通过前十个测试点
最优算法：考虑操作中存在空操作步骤，因此可以优化第一步操作。复杂度O(nm)。也就是下面所列的算法。

//#include <bits/stdc++.h>
#include <stdio.h>
#include <utility>
#include <math.h>
#include <algorithm>
#include <string.h>
using namespace std;

// ================= 代码实现开始 =================

typedef pair<int, int> pii;
#define fi first
#define se second

const int N=2003;

//mind:mind[i][j] 表示从初始状态到达状态（i，j）需要的最少步数
//q：用数组模拟队列
//qh：队头下标
//qt：队尾下标

int mind[N][N];
pii q[N*N];
int qh,qt;

//倒水函数，表示状态p经过第k中策略倒水后的状态
//p：初始状态
//k:策略（在下面进行详细阐述）
//n,m:两杯子容量
//返回值:最终状态

pii to(pii p, int k, int n, int m){
    if(k == 0) //倒空杯子一
        return pii(0, p.se);
    else if(k==1) //倒空杯子二
        return pii(p.fi, 0);
    else if(k==2) //倒满杯子一
        return pii(n, p.se);
    else if(k==3) //倒满杯子二
        return pii(p.fi, m);
    else if(k==4) //将杯子二的水向杯子一倒
        return (n-p.fi < p.se) ? pii(n,p.fi+p.se-n): pii(p.fi+p.se,0); //写法二:pii( min(p.fi+p.se,n), max(0,p.fi+p.se-n) ) 
    else if(k==5) //将杯子一的水向杯子二倒
        return (m-p.se < p.fi) ? pii(p.fi+p.se-m,m): pii(0,p.fi+p.se); //写法二:pii( max(0,p.fi+p.se-m), min(p.fi+p.se,m) ) 
    else
        return p;
}

// 计算答案的函数
// n, m, t, d：意义均与题目描述一致
// 返回值：即为答案
int getAnswer(int n, int m, int t, int d) {
    //初始化，清空队列，将mind所有位置为 -1 表示未访问
    memset(mind,-1,sizeof(mind));
    qh = qt = 0;
    q[++qt] = pii(0,0); //初始化两个水杯容量，均为空。
    mind[0][0] = 0;

    // 进行BFS
    while(qh<qt){
        pii u = q[++qh]; //取出队头元素
        if(mind[u.fi][u.se]==t)
            break;//如果已经进行了t步，那么没必要继续搜索了，退出循环即可
        for(int k=0; k<6; ++k){     //枚举6种策略
            pii v = to(u,k,n,m);
            if( mind[v.fi][v.se]!=-1 ) //判断目标状态是否曾经到达过
                continue;
            q[++qt] = v; //加入队列
//            mind[v.fi][v.se] = qh; //记录mind的倒水次数
            mind[v.fi][v.se] = mind[u.fi][u.se]+1;
        }
    }

    int ans = d;
    for(int i = 0; i<=n; ++i)
        for(int j=0; j<=m; ++j)
            if(mind[i][j] != -1)
                ans = min(ans, abs(i+j-d) );
    return ans;
}

// ================= 代码实现结束 =================

int main() {
    int n, m, t, d;
    scanf("%d%d%d%d", &n, &m, &t, &d);
    int ans = getAnswer(n, m, t, d);
    printf("%d\n", ans);
    return 0;
}