[CF2B]The least round way

2B:The least round way

题意简述

给出一个$n \ast n$的矩阵，每个位置有一个数$a_{i,j}$
一开始在$(1,1)$，只能往右或往下走,最后要走到$(n,n)$。
求一条路径，使得这条路径上所有数的积的后缀零最少。

数据范围

$1 \leq n \leq 1000$
$0 \leq a_{i,j} \leq 10^9$

思路

考虑质因子。
只有$2$和$5$的乘积才会出现一个后缀零。
可以使用DP。
$f[i][j][0]$表示到$(i,j)$位置$2$的个数；$f[i][j][1]$表示到$(i,j)$位置$5$的个数。
进行两次DP，第一次优先保证$2$的个数最小，其次$5$的个数最小，第二次反之。
状态转移方程$f[i][j]=better(f[i-1][j],f[i][j-1])+a[i][j]$
两次的$min(f[n][n][0],f[n][n][1])$的较小值就是答案。
有一个坑点，$a_{i,j}=0$是可能的。出现这样的情况我们需要把那个位置的$2$和$5$的次数都设为$INF$。最后令$ans=min(ans,。1)$
时间复杂度$O(n^2)$

代码

#include<cstdio>
#include<iostream>
using namespace std;
#define INF 0x3f3f3f3f
int in()
{
    char ch=getchar();
    while (ch<'0'||ch>'9')
        ch=getchar();
    int ret=0;
    while (ch>='0'&&ch<='9')
        ret=ret*10+ch-'0',ch=getchar();
    return ret;
}
int n,tmp,ans=INF,cnt;
int a[1010][1010][2],f[1010][1010][2],g[1010][1010];
char pl[2010];
bool flag;
void getans(int x,int y)
{
    if (x==1&&y==1)
        return;
    if (g[x][y]==0)
    {
        getans(x-1,y);
        pl[cnt++]='D';
    }
    else
    {
        getans(x,y-1);
        pl[cnt++]='R';
    }
}
int main()
{
    n=in();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            tmp=in();
            if (!tmp)
            {
                if (!flag)
                {
                    cnt=0;
                    for (int k=2;k<=j;k++)
                        pl[cnt++]='R';
                    for (int k=2;k<=i;k++)
                        pl[cnt++]='D';
                    for (int k=j+1;k<=n;k++)
                        pl[cnt++]='R';
                    for (int k=i+1;k<=n;k++)
                        pl[cnt++]='D';
                    ans=1;
                    flag=1;
                }
                a[i][j][0]=a[i][j][1]=10000;
            }
            else
            {
                while (tmp%2==0)
                    tmp/=2,a[i][j][0]++;
                while (tmp%5==0)
                    tmp/=5,a[i][j][1]++;
            }
        }
    for (int i=0;i<=n;i++)
        f[i][0][0]=f[i][0][1]=f[0][i][0]=f[0][i][1]=INF;
    f[1][0][0]=f[1][0][1]=0;    
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            if (f[i-1][j][0]<f[i][j-1][0]||f[i-1][j][0]==f[i][j-1][0]&&f[i-1][j][1]<f[i][j-1][1])
            {
                f[i][j][0]=f[i-1][j][0]+a[i][j][0],f[i][j][1]=f[i-1][j][1]+a[i][j][1];
                g[i][j]=0;
            }
            else
            {
                f[i][j][0]=f[i][j-1][0]+a[i][j][0],f[i][j][1]=f[i][j-1][1]+a[i][j][1];
                g[i][j]=1;
            }
    if (min(f[n][n][0],f[n][n][1])<ans)
    {
        ans=min(f[n][n][0],f[n][n][1]);
        cnt=0;
        getans(n,n);
    }
    for (int i=0;i<=n;i++)
        f[i][0][0]=f[i][0][1]=f[0][i][0]=f[0][i][1]=INF;
    f[1][0][0]=f[1][0][1]=0;    
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            if (f[i-1][j][1]<f[i][j-1][1]||f[i-1][j][1]==f[i][j-1][1]&&f[i-1][j][0]<f[i][j-1][0])
            {
                f[i][j][0]=f[i-1][j][0]+a[i][j][0],f[i][j][1]=f[i-1][j][1]+a[i][j][1];
                g[i][j]=0;
            }
            else
            {
                f[i][j][0]=f[i][j-1][0]+a[i][j][0],f[i][j][1]=f[i][j-1][1]+a[i][j][1];
                g[i][j]=1;
            }
    if (min(f[n][n][0],f[n][n][1])<ans)
    {
        ans=min(f[n][n][0],f[n][n][1]);
        cnt=0;
        getans(n,n);
    }
    printf("%d\n%s",ans,pl);
    return 0;
}