颜色的长度 【UVA - 1625】【DP】

题目链接

题目大意

输入两个长度分别为n和m的颜色序列，要求按顺序合并成同一个序列，即每次可以把一个序列开头的颜色放到新序列的尾部。对于每个颜色c来说，其跨度L（c）等于最大位置和最小位置之差。

解题思路

我们用dp[i][j]表示从第一个串中拿了i个字母，从第二个串中拿了j个字母的花销。
用c[i][j]表示截止到第一个串的i位置和第二个串的j位置的时候，已经出现并且还没出现完的字母有多少个。
那么转移方程就为：
dp[i][j] = min (dp[i-1][j] + c[i-1][j] , dp[i][j-1] + c[i][j-1]);
memset清零会超时… … (我今天已经因为memset超时第二次了QAQ）

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

const int N=5005;
int dp[N][N],c[N][N];
int a1[40],a2[40],b1[40],b2[40];
char s1[N],s2[N];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s1+1);
        scanf("%s",s2+1);
        int len1=strlen(s1+1);
        int len2=strlen(s2+1);
       // memset(a2,0,sizeof(a2));
       // memset(b2,0,sizeof(b2));
        for(int i=0;i<=30;i++)
        {
            a1[i]=0x3f3f3f3f;
            b1[i]=0x3f3f3f3f;
            a2[i]=0;
            b2[i]=0;
        }
        for(int i=1;i<=len1;i++)
        {
            a1[s1[i]-'A'+1]=min(i,a1[s1[i]-'A'+1]);
            a2[s1[i]-'A'+1]=i;
        }
        for(int i=1;i<=len2;i++)
        {
            b1[s2[i]-'A'+1]=min(i,b1[s2[i]-'A'+1]);
            b2[s2[i]-'A'+1]=i;
        }

        //memset(dp,0,sizeof(dp));
       // memset(c,0,sizeof(c));
        for(int i=0;i<=len1;i++)
        {
            for(int j=0;j<=len2;j++)
            {
                if(!i&&!j)
                    continue;
                int aa=0x3f3f3f3f,bb=0x3f3f3f3f;
                if(i)
                    aa=dp[i-1][j]+c[i-1][j];
                if(j)
                    bb=dp[i][j-1]+c[i][j-1];
                dp[i][j]=min(aa,bb);

                if(i)
                {
                    c[i][j]=c[i-1][j];

                    if(a1[s1[i]-'A'+1]==i&&b1[s1[i]-'A'+1]>j)
                        c[i][j]++;
                    if(a2[s1[i]-'A'+1]==i&&b2[s1[i]-'A'+1]<=j)
                        c[i][j]--;
                }
                else if(j)
                {
                    c[i][j]=c[i][j-1];

                    if(b1[s2[j]-'A'+1]==j&&a1[s2[j]-'A'+1]>i)
                        c[i][j]++;
                    if(b2[s2[j]-'A'+1]==j&&a2[s2[j]-'A'+1]<=i)
                        c[i][j]--;
                }
            }
        }
        printf("%d\n",dp[len1][len2]);
    }

    return 0;
}