HDU 1518 Square(Dfs)

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1518

 

题意：给你n条边，问你用光这些边能不能组成正方形

 

想到用深搜，但是怎么写出来挺难的。。。。。

 

这里主要是超时问题：其实对于某条边，对于那些用过的和不符合条件的for那里可以不扫。。。。。所以Dfs要增加一个参数index表示从那里开始扫。。。但是换另外一条边的时候index要改回0，因为那些未用的，对上一条边来说

不符合条件的，可能符合这条边的条件。。。。

 

代码：

#include <iostream>

using namespace std;



const int M = 10000;



int save[M];

bool used[M];

int sum;

int l;

int n;

int flag;



void Dfs(int now, int len, int index)//now:第几条边  len:该边现在组成的长度 index:用来优化时间

{

    



    if (now == 5)

    {

        flag = 1;

        return ;

    }



    if (len == l)

    {

        Dfs(now + 1, 0, 0);

        if (flag)//优化时间

        {

            return ;

        }

    }



    for (int i = index; i < n; i++)//从index开始优化时间

    {

        if (!used[i] && save[i] + len <= l)

        {

            used[i] = 1;

            Dfs(now, save[i] + len, i + 1);

            if (flag)//优化时间

            {

                return;

            }

            used[i] = 0;

        }

    }

}



int main()

{

    int t;

    scanf("%d", &t);

    while (t--)

    {

        

        sum = 0;

        scanf("%d", &n);

        for (int i = 0; i < n; i++)

        {

            scanf("%d", save + i);

            sum += save[i];

        }

        

        if (sum % 4 != 0)//简答的优化

        {

            puts("no");

            continue;

        }





        l = sum / 4;



        int i;

        for (i = 0; i < n; i++)//有比边长大的边就不行

        {

            if (save[i] > l)

            {

                break;

            }

        }

        if (i != n)

        {

            puts("no");

            continue;

        }





        memset(used, 0, sizeof(used));

        flag = 0;

        Dfs(1, 0, 0);



        if (flag)

        {

            puts("yes");

        }

        else

        {

            puts("no");

        }

    }

    return 0;

}