算法训练 Bit Compressor （dfs 剪枝）

试题 算法训练 Bit Compressor

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　数据压缩的目的是为了减少存储和交换数据时出现的冗余。这增加了有效数据的比重并提高了传输速率。有一种压缩二进制串的方法是这样的：
　　加粗样式将连续的n个1替换为n的二进制表示（注：替换发生当且仅当这种替换减少了二进制串的总长度）
　　（译者注：连续的n个1的左右必须是0或者是串的开头、结尾）
　　比如：11111111001001111111111111110011会被压缩成10000010011110011。原串长为32,被压缩后串长为17.
　　这种方法的弊端在于，有时候解压缩算法会得到不止一个可能的原串，使得我们无法确定原串究竟是什么。请你写一个程序来判定我们能否利用压缩后的信息来确定原串。给出原串长L，原串中1的个数N，以及压缩后的串。
　　L<=16 Kbytes，压缩后的串长度<=40 bits。
输入格式
　　第一行两个整数L,N，含义同问题描述
　　第二行一个二进制串，表示压缩后的串
输出格式
　　输出"YES"或"NO"或"NOT UNIQUE"（不包含引号）
　　分别表示：
　　YES：原串唯一
　　NO：原串不存在
　　NOT UNIQUE：原串存在但不唯一
样例输入
样例1：
32 26
10000010011110011
样例2：
9 7
1010101
样例3：
14 14
111111
样例输出
样例1：YES
样例2：NOT UNIQUE
样例3：NO

AC代码如下：
代码来自https://blog.csdn.net/yi_qing_z/article/details/88084875，稍有改动

#include<stdio.h>
#include<cstring>
#define maxn 50
int L,l,N;
char data[maxn];
int answer;
 
void dfs(int i,int length,int num_1)		//i是原串第i个字符，length是变换后的串长，num_1是变换后串的1累计个数 
{
    if(i==l)		//最后一个位置 
    {
        dfs(i+1,length+1,num_1+data[i]-'0');
        return;
    }
    if(i>l)		//结束条件 
    {
        if(length==L&&num_1==N)
            answer++;
        return;
    }
    
    if(length>L||num_1>N||answer>=2)	//不符合条件，返回 
        return;
        
    if(data[i]=='0')
    {
        dfs(i+1,length+1,num_1);
        return;
    }
    /**data[i]=='1'**/
    if(data[i-1]=='1')		//之前考虑过 
        return;
        
    long long temp=0;
    if(data[i+1]=='0')
        dfs(i+1,length+1,num_1+1);
    if(data[i+1]=='1'&&data[i+2]!='1')
        dfs(i+2,length+2,num_1+2);
    for(int j=i;j<=l;j++)
    {
        temp*=2;
        temp+=data[j]-'0';
        if(temp+num_1>N||temp+length>L)
            break;
        if(temp>j-i+1&&data[j+1]!='1')		//压缩的串1比原来多，并且下一个位置为0或者末尾 
            dfs(j+1,temp+length,num_1+temp);
    }
}
 
 
int main()
{
    scanf("%d",&L);		//原串长
    scanf("%d",&N);		//原来1的个数 
    scanf("%s",data+1);
    l=strlen(data+1);		//新串长 
    answer=0;
    dfs(1,0,0);
    if(answer==0)
        printf("NO");
    else if(answer==1)
        printf("YES");
    else
        printf("NOT UNIQUE");
    return 0;
}