編輯距離優化--二維數組轉一維數組

之前的編輯距離小練習我一直是用二維數組去記錄的，而經神學弟提醒改用了一維數組去優化，發現果然好了很多。

對比我之前的編輯距離小練的文章，這裏再發一次優化後代碼。

POJ3356 AGTC

如圖是兩次代碼的提交對比：

發現時間減了不少，調用內存減得更多。。。所以優化成功。。。

#include<iostream>
#include<string>
#include<cstring>
using namespace std;
int dp[1001];//一維數組
int main()
{
    int a,b,i,j,c,w,tmp;
    string x,y;
    while(cin>>a>>x>>b>>y)
    {
        for(i=0; i<=b; ++i)//初始化
        {
            dp[i]=i;
        }
        for(i=0; i<a; ++i)
        {
            w=dp[0];
            dp[0]=i+1;
            for(j=0; j<b; ++j)
            {
                tmp=dp[j+1];
                if(x[i]==y[j])
                {
                    dp[j+1]=w;
                }
                else
                {
                    dp[j+1]=min(w,min(dp[j+1],dp[j]));
                    dp[j+1]++;
                }
                w=tmp;
            }
        }
        cout<<dp[b]<<endl;
    }
    return 0;
}

HDU4323 Magic Number

提交結果如下圖：

上面的爲未優化，下面的爲優化後。而且都是G++提交AC。。。之前的提交G++TLE，C++AC，而且耗時較多。主要我之前一直沒加上判斷長度。

代碼一：加上了判斷長度*，沒有數組優化。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
using namespace std;
int dp[1505][1505];
char x[1505][1005],y[1005];
int main()
{
    int a,b,i,j,c,t,k,l,p,s,n,m,cas=0,l1,l2[1505];
    scanf("%d",&t);
    while(t--)
    {
        cas++;
        printf("Case #%d:\n",cas);
        scanf("%d%d",&n,&m);
        for(i=0; i<n; ++i)
        {
            scanf("%s",x[i]);
            l2[i]=strlen(x[i]);
        }
        while(m--)
        {
            s=0;
            scanf("%s%d",y,&k);
            l1=strlen(y);
            for(j=0; j<n; ++j)
            {
                if(abs(l1-l2[j])>k)//*判斷長度
                {
                    continue;
                }
                //memset(dp,0,sizeof(dp));
                for(l=0; l<=l2[j]; ++l)
                {
                    dp[l][0]=l;
                }
                for(l=0; l<=l1; ++l)
                {
                    dp[0][l]=l;
                }
                for(l=0; l<l2[j]; ++l)
                {
                    for(p=0; p<l1; ++p)
                    {
                        c=(x[j][l]!=y[p]);
                        dp[l+1][p+1]=min(dp[l][p]+c,min(dp[l+1][p]+1,dp[l][p+1]+1));
                    }
                }
                if(dp[l2[j]][l1]<=k)
                {
                    s++;
                }
            }
            printf("%d\n",s);
        }
    }
    return 0;
}

代碼二：加上了數組優化。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
using namespace std;
int dp[11];//一維數組
char x[1505][10],y[11];
int main()
{
    int a,b,i,j,c,t,k,l,p,s,n,m,cas=0,l1,l2[1505],tmp,w,kk;
    scanf("%d",&t);
    while(t--)
    {
        cas++;
        printf("Case #%d:\n",cas);
        scanf("%d%d",&n,&m);
        for(i=0; i<n; ++i)
        {
            scanf("%s",x[i]);
            l2[i]=strlen(x[i]);
        }
        while(m--)
        {
            s=0;
            scanf("%s%d",y,&kk);
            l1=strlen(y);
            for(j=0; j<n; ++j)
            {
                if(abs(l1-l2[j])>kk)/長度判斷
                {
                    continue;
                }
                for(l=0; l<=l1; ++l)
                {
                    dp[l]=l;
                }
                for(l=0; l<l2[j]; ++l)
                {
                    w=dp[0];
                    dp[0]=l+1;
                    for(k=0; k<l1; ++k)
                    {
                        tmp=dp[k+1];
                        if(x[j][l]==y[k])
                        {
                            dp[k+1]=w;
                        }
                        else
                        {
                            dp[k+1]=min(w,min(dp[k+1],dp[k]));
                            dp[k+1]++;
                        }
                        w=tmp;
                    }
                }
                if(dp[l1]<=kk)
                {
                    s++;
                }
            }
            printf("%d\n",s);
        }
    }
    return 0;
}

兩次對比，發現自己還是很水啊。。。。