把N!分解成质数幂的乘积

题面http://acm.buaa.edu.cn/contest/184/problem/C/
大意就是把N!分解成质数幂的乘积。

最暴力的方法当然会超时,比如下面这个:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int result[5010];
int main()
{
    int m;
    scanf("%d",&m);
    while(m--)
    {
        memset(result,0,sizeof(result));
        int n,cnt=0;
        scanf("%d",&n);
        for(int i=2;i<=n;i++)
        {
            int t=i;
            for(int j=2;j<=n;j++)
            {
                while(t%j==0)
                {
                    t/=j;
                    result[j]++;
                }
            }
        }
        for(int i=2;i<=n;i++)
        {
            if(result[i]!=0)
            {
                ++cnt;
                if(cnt>1) printf("*");
                if(result[i]==1) printf("%d",i);
                else printf("%d^%d",i,result[i]);
            }
        }
        printf("\n");
    }
}

一想,j没必要从2循环到n啊,由于正在学链表,所以用链表来表达多项式写了一遍,结果还是超时,代码如下:

#include<iostream>
#include<cstdio>
using namespace std;
typedef struct term
{
    int coef;
    int exo;
    struct term *link;
}linknode;
void _insert(linknode *&l,int j,int cnt)
{
    linknode *p=new linknode,*t=l->link,*pre=l;
    while(t!=NULL&&j>t->coef)
    {
        t=t->link;
        pre=pre->link;
    }
    if(t==NULL||j<t->coef)
    {
        p->coef=j;
        p->exo=cnt;
        p->link=pre->link;
        pre->link=p;
    }
    else t->exo+=cnt;
}
void _print(linknode *&l)
{
    linknode *t=l->link;
    int cnt=0;
    while(t!=NULL)
    {
        ++cnt;
        if(cnt>1) printf("*");
        if(t->exo==1) printf("%d",t->coef);
        else printf("%d^%d",t->coef,t->exo);
        t=t->link;
    }
    printf("\n");
}
int main()
{
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int n,cnt=0;
        scanf("%d",&n);
        linknode *l=new linknode;
        l->link=NULL;
        for(int i=2;i<=n;i++)
        {
            int t=i;
            for(int j=2;j<=t;j++)
            {
                int cnt=0;
                while(t%j==0)
                {
                    t/=j;
                    ++cnt;
                }
                if(cnt>0) _insert(l,j,cnt);
            }
        }
        _print(l);
    }
}

没办法,只能玩预处理解决找质数的耗时。首先想到的是枚举2~sqrt(n),虽然AC了,但用了828ms,效率有点低。代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef struct term
{
    int coef;
    int exo;
    struct term *link;
}linknode;
void _insert(linknode *&l,int j,int cnt)
{
    linknode *p=new linknode,*t=l->link,*pre=l;
    while(t!=NULL&&j>t->coef)
    {
        t=t->link;
        pre=pre->link;
    }
    if(t==NULL||j<t->coef)
    {
        p->coef=j;
        p->exo=cnt;
        p->link=pre->link;
        pre->link=p;
    }
    else t->exo+=cnt;
}
void _print(linknode *&l)
{
    linknode *t=l->link;
    int cnt=0;
    while(t!=NULL)
    {
        ++cnt;
        if(cnt>1) printf("*");
        if(t->exo==1) printf("%d",t->coef);
        else printf("%d^%d",t->coef,t->exo);
        t=t->link;
    }
    printf("\n");
}
bool flag[5010];
int main()
{
    flag[2]=flag[3]=1;
    for(int i=4,j;i<=5000;i++)
    {
        int q=sqrt(i);
        for(j=2;j<=q;j++)
            if(i%j==0) break;
        if(j>q) flag[i]=1;
    }
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int n,cnt=0;
        scanf("%d",&n);
        linknode *l=new linknode;
        l->link=NULL;
        for(int i=2;i<=n;i++)
        {
            if(flag[i]) _insert(l,i,1);
            else
            {
                int t=i;
                for(int j=2;j<=t;j++)
                {
                    int cnt=0;
                    while(t%j==0)
                    {
                        t/=j;
                        ++cnt;
                    }
                    if(cnt>0) _insert(l,j,cnt);
                    if(flag[t])
                    {
                        _insert(l,t,1);
                        break;
                    }
                }
            }
        }
        _print(l);
    }
}

受人启发,改进了质数的判定,用所谓的“埃氏筛法”。哎,很久之前看过的,都给忘了。不过时间并没有减少太多,索性又改回用数组来做吧,
结果发现真的不应该用链表的。效果不错,降到167ms。代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define maxn 5010
int result[maxn];
bool flag[maxn];
int main()
{
    flag[0]=flag[1]=1;
    for(int i=2;i<=maxn;i++)
        if(!flag[i])
            for(int j=2*i;j<=maxn;j+=i)
                flag[j]=1;
    int m;
    scanf("%d",&m);
    while(m--)
    {
        memset(result,0,sizeof(result));
        int n,cnt=0;
        scanf("%d",&n);
        for(int i=2;i<=n;i++)
        {
            if(!flag[i]) result[i]++;
            else
            {
                int t=i;
                for(int j=2;j<=t;j++)
                {
                    while(t%j==0)
                    {
                        t/=j;
                        result[j]++;
                    }
                    if(!flag[t])
                    {
                        result[t]++;
                        break;
                    }
                }
            }
        }
        for(int i=2;i<=n;i++)
        {
            if(result[i]!=0)
            {
                ++cnt;
                if(cnt>1) printf("*");
                if(result[i]==1) printf("%d",i);
                else printf("%d^%d",i,result[i]);
            }
        }
        printf("\n");
    }
}

再次受人启发,看到了神算法Orz,这次直接降到39ms,终于可以消停一会了。不多说,直接上代码:

#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 5010
bool flag[maxn];
int biao[maxn];
int main()
{
    int cnt=0;
    flag[0]=flag[1]=1;
    for(int i=2;i<=maxn;i++)
        if(!flag[i])
        {
            biao[++cnt]=i;
            for(int j=2*i;j<=maxn;j+=i)
                flag[j]=1;
        }
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;biao[i]<=n;i++)
        {
            int t=n,num=0;
            while(t)
            {
                num+=t/biao[i];
                t/=biao[i];
            }
            if(i>1) printf("*");
            if(num==1) printf("%d",biao[i]);
            else printf("%d^%d",biao[i],num);
        }
        printf("\n");
    }
}

posted on 2015-04-06 16:06  windrises  阅读(499)  评论(0编辑  收藏  举报

导航