UVA1635-唯一分解定理的基本应用2

原题：https://vjudge.net/problem/UVA-1635

这是一个极其典型的“从素因子角度出发”的题目，下面是我的代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 1e5 + 10;
int md[100],mz[100];
int ans[maxn];
int n,m,lenth;
int c[100];

bool check(int n,int i)
{
    int x=n-i;
    int y=i;
    for(int k=0;k<lenth;k++)
    {
        int p=md[k];
        while(x%p==0)
        {
            x/=p;
            c[k]++;
        }
        while(y%p==0)
        {
            y/=p;
            c[k]--;
        }//上下两个循环不能合并,因为这是一个递增积累过程,
    }//也正因如此才需要额外开辟一个数组c.
    for(int k=0;k<lenth;k++)
        if(c[k]<mz[k])return false;
    return true;
}

int factor()
{//用来分解m
    memset(md,0,sizeof(md));
    memset(mz,0,sizeof(mz));
    int c=m,cur=0;
    int p=sqrt(c);
    for(int i=2;i<=p;i++)
    {
        bool ok=false;
        while(c%i==0)
        {
            md[cur]=i;
            mz[cur]++;
            c/=i;
            ok=true;
        }
        if(ok)cur++;
        if(c==1)break;
    }
    if(c>1)md[cur]=c,mz[cur]=1,cur++;
    return cur;
}

int main()
{
    //freopen("input.txt","r",stdin);
    //freopen("text.txt","w",stdout);
    while(cin>>n>>m)
    {
        lenth=factor();//一开始忘记了记录lenth
        int cnt=0;
        memset(c,0,sizeof(c));
        //
        for(int i=1;i<n-1;i++)
            if(check(n,i))ans[cnt++]=i+1;
        printf("%d
",cnt);
        for(int i=0;i<cnt;i++)
        {
            if(i>0)printf(" ");
            printf("%d",ans[i]);
        }
        printf("
");
    }
    return 0;
}