OJ【3099】递归回溯？

题目链接:OJ【3099】递归回溯？

题意:模拟题 叠巧克力 一块巧克力下面最多压两块 上面可以被多块巧克力压 按编码顺序输出对应巧克力下面压的巧克力总和 注意 是按编码顺序输出

分析:递归回溯 递归回溯？……这东西还没咋研究 或许也能做出来 可惜不会= =、我的思路是无脑记录每块巧克力下面压着的巧克力 不存在则存0 重点是后面的处理 我选择递归不断往下查找 然后就是不断的WA 从++blog里翻出一组数据

6
1 2 2 3
2 1 4
3 1 4
4 2 5 6
5 0
6 0

     1

  2     3

      4

  5      6

就是这种情况 然后发现递归的时候4往下加了两次 因为2跟3下面都有4 然后就开始奇思妙想了 总之试了许多种方法。。就不一一列举了……最后想到用标识数组 递归的时候更改标识 同时每次判断 这样就排除了计算每个巧克力下面巧克力总数的时候同一巧克力加两遍的情况 主要想吐槽的是 以为按照输入顺序输出 结果死活想不出来要了后台数据 一对比…………发现自己想太多 按照编码从小到大输出便可 还害自己多创建了个记录输入编码顺序的数组 为自己醉了 话不多说

代码如下:

#include 
   
   
    
    

using namespace std;

struct cho//构建巧克力结构体 记录对应编号巧克力下压的巧克力编号
{
    int down1;
    int down2;
}ch[1001];

int k[1001];//标识数组 标识为0则未计入 不为0则已计算过

void init(int n)//重置标识数组
{
    int i;
    for(i=1;i<=n;i++)
    {
        k[i]=0;
    }
}

int su(int und)//递归计算对应巧克力下压的巧克力(每次计算时标识++ 对应标识不为0时不计入(防重复))
{
    if(ch[und].down1==0&&ch[und].down2==0)
    {
        k[und]++;
        return 0;
    }
    else if(ch[und].down1!=0&&ch[und].down2==0)
    {
        if(k[ch[und].down1]!=0)
            return 0;
        k[ch[und].down1]++;
        return 1+su(ch[und].down1);
    }
    else
    {
        if(k[ch[und].down1]==0&&k[ch[und].down2]==0)
        {
            k[ch[und].down1]++;
            k[ch[und].down2]++;
            return 2+su(ch[und].down1)+su(ch[und].down2);
        }
        if(k[ch[und].down1]!=0&&k[ch[und].down2]==0)
        {
            k[ch[und].down2]++;
            return 1+su(ch[und].down2);
        }
        if(k[ch[und].down1]==0&&k[ch[und].down2]!=0)
        {
            k[ch[und].down1]++;
            return 1+su(ch[und].down1);
        }
        return 0;
    }
}

int main()
{
    ios::sync_with_stdio(false);
    int sum[1001];
    int n,num;
    int i;
    int k;
    while(cin>>n)
    {
        for(i=0;i
    
    
     
     >num;
            sum[num]=0;
            cin>>k;
            if(k==0)
            {
                ch[num].down1=0;
                ch[num].down2=0;
            }
            else if(k==1)
            {
                cin>>ch[num].down1;
                ch[num].down2=0;
            }
            else
            {
                cin>>ch[num].down1>>ch[num].down2;
            }
        }
        for(i=1;i<=n;i++)//处理同时记录每个巧克力下的巧克力总数
        {
            init(n);
            sum[i]+=su(i);
        }

        for(i=1;i<=n;i++)//顺序输出~~~~~注意这里的顺序 不是输入顺序 而是编码顺序TOT
        {
            if(i-1) cout<<' ';
            cout<