直接暴力搜索了,加一点小剪枝就过了,然而题解有DP很强。
题意:
给出N(N≤25)个不同的正整数,求有多少个这样的集合,集合内存在至少两个元素,且任意两个元素差值都大于集合内的元素数量。
正如题面中所描述的,出题人的意图是出一个搜索题。
该题,是按一定顺序枚举子集,然后检查子集是否合法。
时间上,子集数目最多为2N2N 个,判断是否合法可以在搜索过程中检查,因此时间复杂度为O(T*2N2N),T为数据组数。
不过根据题目数据范围xi≤200,合理地搜索一定可以剪枝,因此实际上搜索用时会低于极限复杂度,时间上可以接受。
具体来说:第一步还是先排个序,这样,“任意两个元素差值都大于集合内的元素数量”就等价于“排序后间隔最小的两个相邻元素差值大于集合内的元素数量”;
然后,可以从2开始枚举集合里的元素数目,在确定了元素数目之后dfs依次搜索第i个数,dfs过程中保证第i个数合法。
另外,不枚举集合内元素数目也是可以的,直接dfs依次枚举第i个数,当确定下一个数是否合法时,需要检查截止目前的最小相邻间隔是否大于目前集合内的元素数量。这一过程最好在dfs时记录并传递一个相邻间隔最小值。
值得一提的是,此题可以不用搜索完成,使用动态规划————事实上,在许多时候,指数级复杂度的搜索问题都可以尝试着用动态规划来解决。
其中一种动态规划方法是,定义dp[i][j][k] 表示前i个数里选j个数,并且第i个数要选,相邻最小间隔大于k的方案数。
初始化为dp[i][1][k]=1(1 ≤ i ≤ N, 1 ≤ k ≤ N)
状态转移方程是状态转移方程是,
dp[i][j][k]=∑i−1p=1dp[p][j−1][k](a[i]−[p]>k) dp[i][j][k]=∑p=1i−1dp[p][j−1][k](a[i]−a[p]>k)
答案答案为∑Ni=1∑Nk=2dp[i][k][k] ∑i=1N∑k=2Ndp[i][k][k]∑Ni=1∑Nk=2dp[i][k][k]
dp[i][k][k] 时间复杂度为O(N^4 )。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxl 101
using namespace std;
int n,r,ans;
int a[maxl],b[maxl];
bool yes;
bool cmp(const int &x,const int &y)
{
return x>y;
}
void prework()
{
ans=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
sort(a+1,a+1+n,cmp);
}
void dfs(int last,int k,int len)
{
r=n-(len-k);
for(int i=last+1;i<=r;i++)
{
if(k==1)
{
b[++b[0]]=a[i];
dfs(i,k+1,len);
b[b[0]--]=0;
}
else
if(k==len)
{
if(b[k-1]-a[i]>len)
ans++,yes=true;
}
else
if(b[k-1]-a[i]>len)
{
b[++b[0]]=a[i];
dfs(i,k+1,len);
b[b[0]--]=0;
}
}
}
void mainwork()
{
int rr;
for(int i=2;i<=n;i++)
{
yes=false;rr=n-i+1;
for(int j=1;j<=rr;j++)
{
memset(b,0,sizeof(b));
b[1]=a[j];b[0]=1;
dfs(j,2,i);
}
if(!yes)
return;
}
}
void print()
{
printf("%d\n",ans);
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
prework();
mainwork();
print();
}
return 0;
}