作业训练三编程题11. 数码管

打表题，奖杯来的，建议直接抄

难点在如何读入数据

【问题描述】

液晶数码管用七笔阿拉数字表示的十个数字，把横和竖的一 个短划都称为一笔，即７有３笔，８有７笔等。对于十个数字一种排列，要做到两相邻数字都可以由另一个数字加上几笔或减去几笔组成，但不能又加又减。比如 ７→３是允许的，７→２不允许。任意输入一组数，判断是否符合上述规则，注意，1在右边。

【输入形式】

每行输入一个0~9的排列，数字之间用空格分隔，以-1作为输入结束

【输出形式】

输出YES或NO

【样例输入】

4 1 0 7 3 9 5 6 8 2
3 5 1 6 2 7 9 0 4 8
-1

【样例输出】

YES
NO

#include<bits/stdc++.h> 
using namespace std;

string s;
stringstream ss;
vector<int> a;
map<int,string> checklist = { {1,"1347890"},{2,"28"},{3,"31789"},{4,"4189"},{5,"5689"},{6,"658"},{7,"713890"},{8,"1234567890"},{9,"9134578"},{0,"0178"} };

bool check(int x, int y)
{
	char c = char( '0' + y );
	//string cc = "";
	//cc += c;
	//cout<<c<<" "<<checklist[x]<<endl;
	if( checklist[x].find(c)==-1 ) return false;
	else return true;
}

int main()
{
	while( getline(cin,s) )
	{
		if( s=="-1" ) break;
		
		ss << s;
		int tmp;
		while( ss >> tmp )
		{
			a.push_back(tmp);
		}
		//for(int i=0; i<a.size(); i++) cout<<a[i]<<endl;
		string ans="YES";
		for(int i=1; i<a.size(); i++)
		{
			if( check(a[i-1],a[i]) == false )
			{
				ans="NO";
				break;
			}
		}
		cout<< ans << endl;
		a.resize(0);
		ss.clear();
	}
	return 0;
}