【bzoj3733】[Pa2013]Iloczyn

Description

给定正整数n和k,问能否将n分解为k个不同正整数的乘积

Input

第一行一个数T(T<=4000)表示测试组数
接下来T行每行两个数n(n<=10^9),k(k<=20)

Output

输出T行,若可以被分解,输出”TAK”否则输出”NIE”

Sample Input

3

15 2

24 4

24 5
Sample Output

TAK

TAK

NIE

题解
搜索＋剪枝，先求出所有因数，然后暴力搜索，剪枝就是如果后面最小的k个数当前数相乘大于n就退出。

代码

#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<set>#include<ctime>#include<vector>#include<cmath>#include<algorithm>#include<map>#include<queue>#define mod 1000000007#define ll long long #define N 10000005 #define inf 0x7fffffffusing namespace std;inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int n,k,tot,p[10010];ll f[10010][30];bool dfs(int now,int k,int tmp){    if (!k) return tmp==n;    k--;    for (;now+k<=tot;now++)    {        if (f[now][k]<0) return 0;        if (f[now][k]*tmp>n) return 0;        if (dfs(now+1,k,tmp*p[now])) return 1;    }    return 0;}inline void solve(){    n=read();k=read();tot=0;    for (int i=1;i*i<=n;i++)        if (n%i==0)        {            p[++tot]=i;            if (i*i!=n) p[++tot]=n/i;        }    sort(p+1,p+tot+1);    for (int i=1;i<=tot;i++)    {        ll tmp=1;        for (int j=0;j<k&&i+j<=tot;f[i][j++]=tmp)            if (tmp>0)            {                tmp*=p[i+j];                if (tmp>n) tmp=-1;            }    }    if (dfs(1,k,1)) puts("TAK");else puts("NIE");}int main(){    int Case=read();    while (Case--) solve();    return 0;}