汽车加油行驶问题

#include "iostream"
#include "algorithm"
#include "fstream"
using namespace std;
#define INF 10000
/*
f[i][j][0]表示汽车从网格点(1,1)行驶至网格点(i,j)所需的最少费用
f[i][j][1]表示汽车行驶至网格点(i,j)还能行驶的网格边数

s[i][0]表示x轴方向
s[i][1]表示y轴方向
s[i][2]表示行驶费用

f[i][j][0] = min{f[ i+s[k][0] ][ [j+s[k][1] ][0] + s[k][2]}
f[i][j][1] = f[ i+s[k][0] ][ [j+s[k][1] ][1] - 1;

f[1][1][0] = 0
f[1][1][1] = K
f[i][j][0] = f[i][j][0] + A , f[i][j][1] = K   如果(i, j)是油库
f[i][j][0] = f[i][j][0] + C + A, f[i][j][1] = K  (i, j)不是油库，且f[i][j][1] = 0
*/

int N;  //方形网络规模
int A;  //汽车在行驶过程中遇到油库应加满油并付加油费A
int C;  //在需要时可在网格点处增设油库，并付增设油库费用C（不含加油费A）
int B;  //当汽车行驶经过一条网格边时，如果其x坐标或y坐标减少，应付费用B
int K;  //装满油后，还能行驶K条边

int f[50][50][2];
int s[4][3] = {{-1,0,0},{0,-1,0},{1,0,B},{0,1,B}};

int a[50][50];  //方形网络

int dyna()
{   
    int i, j, k;
    for (i=1;i<=N;i++)
    {
        for (j=1;j<=N;j++)
        {
            f[i][j][0]=INF;
            f[i][j][1]=K;
        }
    }

    f[1][1][0] = 0;
    f[1][1][1] = K;

    int count = 1;
    int tx, ty;
    while(count)
    { 
        count = 0;
        for(i=1; i<=N; i++)
        {
            for(int j=1; j<=N; j++)
            {
                if(i==1 && j==1)
                    continue;
                int minStep = INF;
                int minDstep;
                int step, dstep;
                for(k=0; k<4; k++) //可走的四个方向
                {
                    tx = i + s[k][0];
                    ty = j + s[k][1];
                    if(tx<1 || ty<1 || tx>N || ty>N)  //如果出界
                        continue;

                    step =  f[tx][ty][0] + s[k][2]; 
                    dstep = f[tx][ty][1] - 1;
                    if(a[i][j] == 1)  //如果是油库
                    {
                        step += A;
                        dstep = K; 
                    }
                    if(a[i][j]==0 && dstep == 0 && (i!=N||j!=N))  //如果不是油库,且油已经用完
                    {
                        step += A + C;
                        dstep = K;
                    }
                    if(step < minStep)  //可以走
                    {
                        minStep = step;
                        minDstep = dstep; 
                    }
                }               

                if(f[i][j][0] > minStep)  //如果有更优解
                {
                    count++;
                    f[i][j][0] = minStep;
                    f[i][j][1] = minDstep;
                }
            }
        }
    }
    return f[N][N][0];
}

int main()
{
    ifstream fin("car.txt");
    cout  << "输入方格规模：";
    fin >> N; cout << N;
    cout << "\n输入装满油后能行驶的网格边数：";
    fin >> K; cout << K;
    cout << "\n输入加油费：";
    fin >> A; cout << A;
    cout << "\n输入坐标减少时应付的费用：";
    fin >> B; cout << B; s[2][2] = s[3][2] = B;
    cout << "\n输入增设油库费用：";
    fin >> C; cout << C;
    cout << "\n输入方形网络：\n";
    for(int i=1; i<=N; i++)
    {
        for(int j=1; j<=N; j++)
        {
            fin >> a[i][j];
            cout << a[i][j] << " ";
        }
        cout  << endl;
    }

    cout << "最优行驶路线所需的费用为：" << dyna() << endl;
    fin.close();
    return 0;
}