给自己出题的小X
Description:
小X学习了dfs,为了练习搜索,开始给自己出题玩。
玩着玩着,一会把自己难住了,一会又被自己难倒了,真是有趣诶!
小X出的题:
现在有N个不同的正整数,求它们可以组成多少个这样的集合,满足:
•集合内的元素数量S>1
•集合内任意两个数的差的绝对值都大于集合内的元素数量。
Input
第一行,一个正整数T(T<=20)表示数据组数。
对于每组数据,有两行。第一行为一个正整数N(3≤N≤25),第二行为N个用空格隔开的正整数xi(xi≤200)。
Output
对于每组数据,输出一行一个整数表示题中所描述的集合的个数。
Sample Input
1
5
2 3 5 8 1
Sample Output
6
Hint
{5,8},{1,5},{1,8},{2,5},{2,8},{3,8}.
题目大意:
可以理解成对一个集合,去选择满足要求的子集。其中要求子集中任意两个元素的差的绝对值大于元素个数,并且元素个数至少为2。
解题思路:
1.感觉元素个数不是很多,采用暴力DFS搜索(其实是看题意感觉应该就是用DFS吧…也不知道有没有更强力的方法…)
2.为了简化后序搜索,先对元素进行排序。这样做同时也就确定了大小,相当于顺带去掉了绝对值运算。
3.我这里做法的意图是每次去找下一个可行的数,一直找到还差一个达到子集个数要求的时候,求出最后一个元素的可行个数。期中前面的搜索要对最小满足的下标以后的数都遍历。
具体做法不太好讲,看注释吧…其中fun(int now_num, int now_pos)函数是计算 比第now_pos个元素 大超过now_num的 下一个元素的位置。如果没有就返回0。
核心代码:
int dfs(int now_num, int ser_pos, int count) { // now_num表示当前测试的子集大小,ser_pos表示当前搜索的点,cout表示搜到并加入的点(由于一开始的起点自动加入,所以集合找全的的时候count值为now_num-1)
if (count == now_num - 2) { //当count为now_num-2的时候表示还差一个数就达到了子集要求,此时去找最后一个数的最小下标。
int next = fun(now_num, ser_pos);
if (next) //因为数组是有序的,找到最近下标以后,之后的元素是都满足要求的
return n - next;
else
return 0;
}
else { //当前的子集元素还差2个以上
int next = fun(now_num, ser_pos);
if (next) { //找到下一个满足条件的最小位置
int temp = 0;
for (int i = next; i < n; i++) { //注意:要遍历满足当前步的全部元素
temp += dfs(now_num, i, count + 1);
}
return temp;
}
else //加入了count个元素以后找不到满足要求的下一个元素了
return 0;
}
}
3.先遍历子集大小,再对开始位置进行遍历。
源代码:
#include<iostream>
#include<stdio.h>
#include<set>
#include<algorithm>
#include<vector>
using namespace std;
int a[205];
int n, re;
int fun(int now_num, int now_pos) {
if (now_pos >= n)
return 0;
int i = now_pos + 1;
while (i < n&&a[i] - a[now_pos] <= now_num)
i++;
if (i<n)
return i;
else
return 0;
}
int dfs(int now_num, int ser_pos, int count) {
if (count == now_num - 2) {
int next = fun(now_num, ser_pos);
if (next)
return n - next;
else
return 0;
}
else {
int next = fun(now_num, ser_pos);
if (next) {
int temp = 0;
for (int i = next; i < n; i++) {
temp += dfs(now_num, i, count + 1);
}
return temp;
}
else
return 0;
}
}
int main() {
int times;
scanf("%d", ×);
for (int ti = 0; ti < times; ti++) {
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d", &a[i]);
sort(a, a + n);
re = 0;
for (int i = 2; i <= n; i++) {
for (int j = 0; j<=n-i; j++)
re += dfs(i, j, 0);
}
printf("%d\n", re);
}
return 0;
}