刷题记录:牛客NC13585安卓图案解锁

传送门:牛客

栗主席(lizi)是某xxxx大学的一个不得了的程序猿，然而没想到吧，他竟然有女盆友，我们假设为QAQ！！！
那天，QAQ问栗子：你的小米5s的图像解锁密码到底是多少？
栗子：嘛？我仔细想想...   
QAQ：你仿佛在逗我...
...
栗子：我的图像解锁用过好多次密码，后来都是用指纹解锁，所以忘记密码辣。但是我记得可能是那几个密码
QAQ：那你务必告诉我...
栗子: ...
然后，栗子就写下了一堆可能的密码，安卓图案解锁中，数字对应的位置已经标出。
但是栗子当然不想把真正的密码告诉QAQ，所以给QAQ的一系列的密码中，甚至有一些密码，是不符合安卓图案解锁的规则的。
QAQ也知道栗子肯定不老实，给了很多错的密码，甚至不符合规则的密码，所以想请你来找出，哪些密码是不符合规则的。
安卓图案解锁的密码有这样的一些特点：
1.每个数字最多只会被使用一次。
2.如果想直接连接两个数字，但是线段中会经过另一个数字，当且仅有那个数字已经在之前就被使用过了，才会合法。(比如你想从1直接连接到9，那么要么是1->3->9,要么是3在之前已经被使用过了，然后才能直接从1->9)
输入:
14569
1953
15963
15953
输出:
YES
NO
YES
NO

主要思路:最长的符合字符串的长度也应该小于等于9,所以长度方面可以直接特判
观察题目,我们需要判断的是连续两个字符如果恰好隔一个数字的话(且中间那个数字之前没有遍历过)就是不符合的
这时候我们的重点就是判断上面这个情况,如果此时我们使用数组之类的去储存两个数字之间的关系的话会很麻烦(实现起来很麻烦),此时我们换个思路,我们将不能放在一起的数字都直接赋值上同一个值,比如1,3,7,9这四个数字两两之间都是间隔一个数字的,也就是符合上述情况,此时我们的判断就会很方便且容易实现了

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
string s;
int vis[20];
int to_num[20]={0,1,2,1,3,4,3,1,2,1};
int main() {
	while((cin>>s)) {
		int flag=0;
		memset(vis,0,sizeof(vis));
		if(s.length()>9){
			cout<<"NO"<<endl;
			continue;
		} 
		vis[s[0]-'0']=1;
		for(int i=1;i<s.length();i++){
			int num=s[i]-'0';
			if(vis[num]==1) {
				cout<<"NO"<<endl;
				flag=1;
				break;
			}else {
				int num2=s[i-1]-'0';
				if(to_num[num]==to_num[num2]) {
					if(vis[(num+num2)/2]!=1) {
						cout<<"NO"<<endl;
						flag=1;
						break;
					}
				}
				vis[num]=1;
			}
		}
		if(flag==0) cout<<"YES"<<endl;
	}
	return 0;
}