NYOJY 491 幸运三角形

描述

话说有这么一个图形，只有两种符号组成（‘+’或者‘-’），图形的最上层有n个符号，往下个数依次减一，形成倒置的金字塔形状，除第一层外（第一层为所有可能情况），每层形状都由上层决定，相邻的符号相同，则下层的符号为‘+’，反之，为‘-’;如下图所示（n = 3 时的两种情况)：

如果图中的两种符号个数相同，那这个三角形就是幸运三角形，如上图中的图（2）.

输入

有多组测试数据（少于20组）。
每行含一个整数n（0<n<20)。

输出

输出相应的幸运三角形个数。

样例输入

3
4

样例输出

4
6

///排列组合第一行，然后直接寻找是否相等
#include <iostream>
#include <cstdio>
#include <cstring>
#define N 1010
#define MOD 1000000007
using namespace std;
int n,m,ans[];
int a[];///待排列的数存储在此
bool vis[];
int p[N][N],k;
int sum,num,tot;
void dfs(int cnt)///按字典序输出n个数选m个数的所有排列
{
    if(cnt==m)
    {
        for(int i=; i<m; i++)
        {
            if(ans[i] == ans[i+] && i+ < m)
                p[][k] = ,sum++;
            else
                p[][k] = ;
            k++;
            if(ans[i] == )
                sum++;
        }
        int c = ;
        while(c < m)
        {
            k = ;
            for(int i = ; i < m - c + ; i++)
            {
                if(p[c-][i] == p[c-][i+] && i+ < m - c + )
                    p[c][k] = ,sum++;
                else
                    p[c][k] = ;
                k++;
            }
            c++;
        }
        if(sum == num)
            tot++;
        k = ;
//        for(int i=0; i<m; i++)
//            cout<<ans[i]<<"***";
//        cout<<endl;
//        for(int i = 1; i < m; i++)
//        {
//            for(int j = 0 ; j < m-i; j++)
//                cout<<p[i][j]<<"***";
//            cout<<endl;
//        }
//        cout<<sum<<" "<<num<<endl;
        sum = ;
        return ;
    }
    for(int i=; i<n; i++)
    {
        ans[cnt]=a[i];
        dfs(cnt+);
    }
}
int main()
{
    while(cin>>m)
    {
        k = ;
        n = ;
        sum = ;
        tot = ;
        num = m * (m + );
        if(num % )
        {
            cout<<<<endl;
            continue;
        }
        else
            num /= ;
        fill(vis,vis+,);
        a[] = ;
        a[] = ;
        dfs();
        cout<<tot<<endl;
    }
    return ;
}

TLE

纯粹dfs,回溯还未入门，不过如果应试的话完全可以打表，毕竟数据量不是太大。

///排列组合第一行，然后直接寻找是否相等
#include <iostream>
#include <cstdio>
#include <cstring>
#define N 1010
#define MOD 1000000007
using namespace std;
int n,m,ans[];
int a[];///待排列的数存储在此
bool vis[];
int p[N][N],k;
int sum,num,tot;
void dfs(int cnt)///按字典序输出n个数选m个数的所有排列
{
    if(cnt==m)
    {
        for(int i=; i<m; i++)
        {
            if(ans[i] == ans[i+] && i+ < m)
                p[][k] = ,sum++;
            else
                p[][k] = ;
            k++;
            if(ans[i] == )
                sum++;
        }
        int c = ;
        while(c < m)
        {
            k = ;
            for(int i = ; i < m - c + ; i++)
            {
                if(p[c-][i] == p[c-][i+] && i+ < m - c + )
                    p[c][k] = ,sum++;
                else
                    p[c][k] = ;
                k++;
            }
            c++;
        }
        if(sum == num)
            tot++;
        k = ;
//        for(int i=0; i<m; i++)
//            cout<<ans[i]<<"***";
//        cout<<endl;
//        for(int i = 1; i < m; i++)
//        {
//            for(int j = 0 ; j < m-i; j++)
//                cout<<p[i][j]<<"***";
//            cout<<endl;
//        }
//        cout<<sum<<" "<<num<<endl;
        sum = ;
        return ;
    }
    for(int i=; i<n; i++)
    {
        ans[cnt]=a[i];
        dfs(cnt+);
    }
}
int main()
{
    for(m = ; m < ; m++)
        //while(cin>>m)
    {
        k = ;
        n = ;
        sum = ;
        tot = ;
        num = m * (m + );
        if(num % )
        {
            cout<<<<",";
            continue;
        }
        else
            num /= ;
        fill(vis,vis+,);
        a[] = ;
        a[] = ;
        dfs();
        cout<<tot<<",";
    }
    return ;
}

打表过了

回溯才是正解，可以减掉重复计算的值，但是自己没搞懂，留下正解先

///边回溯边计算，避免重复计算
#include"iostream"
#include<cstring>
#include<stdio.h>
#include<time.h>
using namespace std;
typedef unsigned char uchar;

//char cc[2]={'+','-'};    //便于输出
int n,                     //第一行符号总数
    half,                 //全部符号总数一半
    counter;             //1计数，即  '-' 号计数

char **p;                //符号存储空间
long sum;                //符合条件的三角形计数

//t，第一行第 t个符号
void Backtrace(int t)
{
    int i, j;

    if( t > n )
        sum++;
    else
    {
       for(i=; i<; ++i)  //只取  0('+')  或者  1('-')
       {
            p[][t] = i;   //第一行第 t个符号
            counter += i;        //'-'号统计
            for(j=; j<=t; ++j)    //当第一行符号 >=2时，可以运算出下面行的某些符号（第一行有几个数就可以相应往下计算几行，每次计算过的就不用重复计算）
            {
                p[j][t-j+] = p[j-][t-j+]^p[j-][t-j+];//通过异或运算下行符号
                counter += p[j][t-j+];
            }
            if( (counter <= half) && ( t*(t+)/ - counter <= half) )//若符号统计未超过半数，并且另一种符号也未超过半数
                Backtrace(t+);            //在第一行增加下一个符号
            //回溯，判断另一种符号情况
            for(j=; j<=t; ++j)
                counter -= p[j][t-j+];
            counter -= i;
       }
    }
}

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        counter = ;
        sum = ;
        half = n*(n+)/;

        if( half% ==  )//总数须为偶数，若为奇数则无解
        {
            half /= ;
            p = new char *[n+];
            for(int i=; i<=n; ++i)
            {
               p[i] = new char[n+];
               memset(p[i], , sizeof(char)*(n+));
            }
            Backtrace();
        }
        printf("%d\n", sum);
    }
    return ;
}

回溯正解

http://www.cnblogs.com/dongsheng/archive/2013/06/16/3138607.html