bzoj 3733: [Pa2013]Iloczyn【dfs】

参考:http://www.cnblogs.com/clrs97/p/5125976.html
瞎搞约数失败...滚去搜索
dfs(x,y,z) 表示当前可选第x到第m个约数,还要选y个约数,已有z的乘积,搜的时候减掉大于n的情况

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=2005;
int n,k,tot,q[N],f[N][35];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
bool dfs(int x,int y,int z)
{
	if(!y)
		return z==n;
	for(y--;x+y<=tot;x++)
	{
		if(f[x][y]<0)
			return 0;
		if(1LL*f[x][y]*z>n)
			return 0;
		if(dfs(x+1,y,z*q[x]))
			return 1;
	}
	return 0;
}
int main()
{
	int T=read();
	while(T--)
	{
		n=read(),k=read(),tot=0;
		for(int i=1;i*i<=n;i++)
			if(n%i==0)
			{
				q[++tot]=i;
				if(i*i!=n)
					q[++tot]=n/i;
			}
		sort(q+1,q+tot+1);
		for(int i=1;i<=tot;i++)
		{
			long long t=1;
			for(int j=0;j<k&&i+j<=tot;f[i][j++]=t)
				if(t>0)
				{
					t*=q[i+j];
					if(t>n)
						t=-1;
				}
		}
		puts(dfs(1,k,1)?"TAK":"NIE");
	}
	return 0;
}
posted @ 2018-04-26 17:34  lokiii  阅读(185)  评论(0编辑  收藏  举报