计算多项式Poj(1996)

题目链接:http://poj.org/problem?id=1996

思路：

刚开始打了个二维表，调了一个小时，爆内存了。

#include <stdio.h>
#include <string.h>
 
int a[];
int y[];
int ans[];
int dp[][];
 
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,,sizeof(a));
        memset(y,,sizeof(y));
        memset(ans,,sizeof(ans));
        memset(dp,,sizeof(dp));
 
        int m,n;
        scanf("%d%d",&m,&n);
        for(int i=; i<=m; i++)
        {
            scanf("%d",&a[i]);
            if(i==)
                ans[i] = a[i];
        }
        for(int i=; i<=n; i++)
        {
            scanf("%d",&y[i]);
            dp[][i] = y[i];
        }
 
        int pos=n;
        for(int i=; i<=n; i++)
            printf("%d ",dp[][i]);
        puts("");
        for(int k=; k<=m; k++)
        {
            for(int i=; i<=n; i++)
            {
                for(int j=; j<=pos; j++)
                {
                    dp[k][i+j]+=dp[k-][j]*y[i];
                    //printf("%d ",dp[k][i+j]);
                }
            }
            pos+=n;
            //for(int i=0; i<=pos; i++)
                //printf("%d ",dp[k][i]);
            //puts("");
        }
 
        pos=n;
        for(int i=; i<=m; i++)
        {
            for(int j=; j<=pos; j++)
            {
                dp[i][j] = dp[i][j] * a[i];
                //printf("%d ",dp[i][j]);
            }
            //puts("");
            pos+=n;
        }
 
        for(int i=; i<=n*m; i++)
        {
            for(int j=; j<=m; j++)
                ans[i] +=dp[j][i];
        }
        for(int i=; i<=n*m-; i++)
            printf("%d ",ans[i]);
        printf("%d",ans[n*m]);
        puts("");
    }
    return ;
}

然后压缩了一下。

#include <stdio.h>
#include <string.h>
 
int a[];
int b[];
int dp[];
int ans[];
int tmp[];
 
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,,sizeof(a));
        memset(b,,sizeof(b));
        memset(dp,,sizeof(dp));
        memset(ans,,sizeof(ans));
 
        int m,n;
        scanf("%d%d",&m,&n);
 
        for(int i=;i<=m;i++)
            scanf("%d",&a[i]);
        ans[] = a[];
 
        for(int i=;i<=n;i++)
        {
            scanf("%d",&b[i]);
            dp[i] = b[i];
        }
        if(m!=)
        {
            for(int i=;i<=n;i++)
                ans[i] +=a[]*dp[i];
        }
        int pos = n;
        for(int k=;k<=m;k++)
        {
            memset(tmp,,sizeof(tmp));
            for(int i=;i<=n;i++)
            {
                for(int j=;j<=pos;j++)
                    tmp[i+j] +=b[i]*dp[j];
            }
 
            pos+=n;
 
            for(int i=;i<=pos;i++)
                dp[i] = tmp[i];
 
            for(int i=;i<=pos;i++)
                ans[i]+=a[k]*dp[i];
        }
        for(int i=;i<=pos-;i++)
            printf("%d ",ans[i]);
        printf("%d\n",ans[pos]);
 
    }
    return ;
}

现在还不知道WA在哪里。 打死POJ!!!

后来借鉴了帆哥的代码，一模一样的思路，就AC了。

要是哪个大神看出错了，记得@我啊！

#include<iostream>
using namespace std;
#define INF 10005
int x[INF],y[INF],z[INF],r[INF],p[INF];
int t;
int n,m;
int size;
void calc(int k)
{
    int i,j,h;
    if(k==)
    {
        z[k]+=x[k];
        size=;
    }
    else if(k==)
    {
        for(i=;i<=m;i++)
        {
            z[i]+=(x[k]*y[i]);
        }
        size=m;
    }
    else
    {
 
            memset(p,,sizeof(p));
            for(j=;j<=m;j++)
            {
                for(h=;h<=size;h++)
                {
                    p[j+h]+=(y[j]*r[h]);
                }
            }
            size+=m;
            for(j=;j<=size;j++)
            {
                r[j]=p[j];
            }    
 
            for(i=;i<=size;i++)
            {
                z[i]+=(r[i]*x[k]);
            }
 
    }
 
}
int main()
{
    int i,j;
    scanf("%d",&t);
    while(t--)
    {
        memset(x,,sizeof(x));
        memset(y,,sizeof(y));
        memset(z,,sizeof(z));
        memset(r,,sizeof(r));
        memset(p,,sizeof(p));
        scanf("%d%d",&n,&m);
        for(i=;i<=n;i++)
        {
            scanf("%d",&x[i]);
        }
        for(i=;i<=m;i++)
        {
            scanf("%d",&y[i]);
            r[i]=y[i];
        }
 
        for(i=;i<=n;i++)
        {
            calc(i);
        }
 
        for(i=;i<size;i++)
        {
            printf("%d ",z[i]);
        }
        printf("%d",z[size]);
        printf("\n");
    }
 
}