HDU 1518 Square(回溯搜索,减枝很巧妙啊)

题目地址:点击打开链接

题意:给你n个棍子,这些棍子的长度告诉你,问把这个棍子全都用上,能否组成一个正方形

思路:用回溯一般就是求可能情况(这种情况能否组成),所有组成情况的最值,能组成这种情况的个数(八皇后问题),这道题的减枝想到了还容易wrong,注意看我注释掉的部分,减枝比较容易想,每次只找比它长的,因为组成一条正方形的边的长度肯定有长有短,而这道题只求可能情况,所以这种减枝方法可行

AC代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>

using namespace std;

int a[25];
int visit[25];
int edge;
int m;
bool flag;

void dfs(int start,int sum,int now)
{
    int i;
    if(flag)
        return;
    if(sum == 4)
    {
        flag = true;
        return;
    }
    for(i=start; i<m; i++)
    {
        if(!visit[i])
        {
            visit[i] = 1;
            if(now + a[i] == edge)
            {
                //dfs(i+1,sum+1,0);
                dfs(0,sum+1,0);//组成一条边的时候要从头开始搜
            }
            else if(now + a[i] < edge)
            {
                dfs(i+1,sum,now+a[i]);
            }
            visit[i] = 0;
        }
    }
}

int main()
{
    int t,i;
    scanf("%d",&t);
    while(t--)
    {
        flag = false;
        int sum = 0;
        memset(visit,0,sizeof(visit));
        scanf("%d",&m);
        for(i=0; i<m; i++)
        {
            scanf("%d",&a[i]);
            sum += a[i];
        }
        if(sum % 4 != 0)
        {
            printf("no\n");
            continue;
        }
        edge = sum / 4;
        for(i=0; i<m; i++)
        {
            if(a[i] > edge)
                break;
        }
        if(i != m)
        {
            printf("no\n");
            continue;
        }
        sort(a,a+m);
        dfs(0,0,0);
        if(flag)
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}

超时代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>

using namespace std;

int a[25];
int visit[25];
int edge;
int m;
bool flag;

void dfs(int sum,int now)
{
    int i;
    if(flag)
        return;
    if(sum == 4)
    {
        flag = true;
        return;
    }
    for(i=0; i<m; i++)
    {
        if(!visit[i])
        {
            visit[i] = 1;
            if(now + a[i] == edge)
            {
                dfs(sum+1,0);
            }
            else if(now + a[i] < edge)
            {
                dfs(sum,now+a[i]);
            }
            visit[i] = 0;
        }
    }
}

int main()
{
    int t,i;
    scanf("%d",&t);
    while(t--)
    {
        flag = false;
        int sum = 0;
        memset(visit,0,sizeof(visit));
        scanf("%d",&m);
        for(i=0; i<m; i++)
        {
            scanf("%d",&a[i]);
            sum += a[i];
        }
        if(sum % 4 != 0)
        {
            printf("no\n");
            continue;
        }
        edge = sum / 4;
        for(i=0; i<m; i++)
        {
            if(a[i] > edge)
                break;
        }
        if(i != m)
        {
            printf("no\n");
            continue;
        }
        dfs(0,0);
        if(flag)
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}


你可能感兴趣的:(HDU 1518 Square(回溯搜索,减枝很巧妙啊))