网易互娱2017在线笔试——题目一：电子数字

电子数字

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

电子数字在生活中很常见，而许多的电子数字是由LED数码管制作而成。数字LED数码管一般由7个发光二极管封装在一起，组成’8’字型，引线在内部连接完成。如下图所示，我们可以对每个发光管进行编码从1到7。而数字0到数字9可以由这七根发光管的亮暗来表示。

对LED数码管的二极管进行编码


用LED数码管表示数字0-9

假设我们现在有从左到右排列好的K个LED数码管，并且我们已知每个数码管当前有哪些编号的二极管是亮着的，另外剩余的二极管由于某些原因，我们并不清楚它们的亮暗情况。由于已经有部分二极管是确定亮着的，所以每个LED数码管能表示的数字范围会有所缩小，譬如假设1号二极管已经确定是亮着的状态，那么这个LED数码管就不能表示数字1和4。

我们想知道的是，给定一个数N，在这K个LED数码管的当前亮暗的状态下，所有可能表示的数中，比N小的数有多少个。

注意，前导0是必须的，假设有4个数码管的话，’0000’表示0，’0123’表示123，即每个数的表示方法唯一。

输入

每个输入数据包含多个测试点。

第一行为测试点的个数 S ≤ 100。之后是 S 个测试点的数据。测试点之间无空行。

每个测试点的第一行为 K(1 ≤ K ≤ 5)和N(0 ≤ N ≤ 109)。之后是K行，每行表示对应数码管已点亮的二极管的情况。每行至少包含一个数字，表示对应点亮的二极管的编号，即每个数码管至少有一根二极管是点亮的。二极管编号的范围保证在1到7之间，且每行无重复编号。

注意表示数码管点亮情况的每行数字之间以及行首行末之间可能存在冗余空格，每行的字符总长度不超过100。

输出

对于每个测试点，对应的结果输出一行，表示这K个数码管在当前状态下，所有可能表示的数中，比N小的数有多少个。

样例解释

第一个样例中，只有’020’, ‘026’, ‘028’符合要求。

第三个样例中，只有’000’符合要求。

样例输入

3
3 50
3  1
  1  4  5  
1   5  6 7
4 100
1 2 3
   4   5
6
  7  
1 1
  7

样例输出

3
0
1

代码如下，已通过AC测试。
原创代码，请勿转载！

#include "bits/stdc++.h"

using namespace std;

set<int> set_1 = {0, 2, 3, 5, 6, 7, 8, 9};
set<int> set_2 = {0, 4, 5, 6, 8, 9};
set<int> set_3 = {0, 1, 2, 3, 4, 7, 8, 9};
set<int> set_4 = {2, 3, 4, 5, 6, 8, 9};
set<int> set_5 = {0, 2, 6, 8,};
set<int> set_6 = {0, 1, 3, 4, 5, 6, 7, 8, 9};
set<int> set_7 = {0, 2, 3, 5, 6, 8, 9};

vector<set<int>> set_sum = {set_1, set_2, set_3, set_4, set_5, set_6, set_7};

map<int, set<int>> &test(int K)
{
    static map<int, set<int>> initial_results;
    //此为静态变量
//静态变量再次使用前一定要清空！
    for (int i = 0; i < K; ++i)
    {
        int x;
        string line3;
        getline(cin, line3);
        istringstream read(line3);
        vector<int> sum = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
        while (read >> x)
        {
            vector<int> ivec(10);
            auto iter = set_intersection(sum.begin(), sum.end(), set_sum[x - 1].begin(), set_sum[x - 1].end(),
                                         ivec.begin());
            ivec.resize((unsigned long) (iter - ivec.begin()));
            //重新确定ivec大小
            sum.clear();
            sum = ivec;
        }
        set<int> sum_set;
        sum_set.insert(sum.begin(), sum.end());
        initial_results[i] = sum_set;
    }
    return initial_results;
}

int determine(string N, map<int, set<int>> &initial)
{
    unsigned long N_length{N.length()}, initial_size{initial.size()};
    if (N_length > initial_size)
    {
        int sizes = 1;
        for (auto it = initial.begin(); it != initial.end(); ++it)
        {
            sizes *= (*it).second.size();
        }
        return sizes;
    } else
    {
        auto it = initial.begin();
//前端数码管多出的位检测是否含0
        {
            if ((*it++).second.count(0) == 0)return 0;
        }
        vector<int> final_result, lower_equal;
        auto it_std = it;
        //it_std指向和给定目标值相对应的最大数字位
        for (int i = 0; it != initial.end(); ++it)
//计算和给定目标值相对应的数字位的数码管小于等于目标值的数字位的值的个数
        {
            int i_sizes = (int &&) distance((*it).second.begin(),
                                            (*it).second.upper_bound(N[i++] - '0'));//up大于lower大于等于
            lower_equal.push_back(i_sizes);
            final_result.push_back((int &&) (*it).second.size());
        }
        it = it_std;
        //以下为处理当前某个数码管可能表示的最大值恰好等于相对应数字未的目标值的情况
        for (int i = 0; it != initial.end(); ++it)//预处理
        {
            final_result[i] = lower_equal[i];
            if (*(*it).second.lower_bound(N[i] - '0') != N[i] - '0')
                break;
            ++i;
        }
        bool control = false;
        do
        {
            it = it_std;
//数码管第一位(与目标值最大位对应)为0匹配不调整
            {
                ++it;
                if (final_result[i] == 0) 
                //发现第一个0匹配时，将前一位匹配数-1，将当前及以后的匹配数置初始结果全值
                {
                    final_result[i - 1] = final_result[i - 1] - 1;
                    for (int j = i; j < final_result.size(); ++j)
                    {
                        final_result[j] = (int) (*it++).second.size();
                    }
                    break;
                }
            }
            for (int i = 1; i < (int) final_result.size(); ++i)
            {
                if (final_result[i] == 0)
                {
                    control = true;
                    break;
                } else control = false;
            }
        } while (control);
        int sizes = 1;
        for (int i = 0; i < final_result.size(); ++i)
        {
            sizes *= final_result[i];
        }
        return sizes;
    }
}

int main()
{
    string line;
    int n;//测试点个数
    getline(cin, line);
    istringstream read_n(line);
    read_n >> n;
    for (int i = 0; i < n; ++i)
    {
        string line1;
        int K;
        string N;
        getline(cin, line1);
        istringstream readTest(line1);
        readTest >> K >> N;
        cout << determine(N, test(K)) << "\n";
    }
}