[HNOI2014]抄卡组

[Luogu3234] [LOJ2208]

题解及代码

锻炼哈希码力的一道题 , 具体细节见代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define Debug(x) cout<<#x<<"="<<x<<endl
using namespace std;
typedef unsigned long long ULL;
const int INF=1e9+7;

const int N=1e5+5;
const int M=1e7+5;
const int base=2333;

ULL pow[M];//使用unsigned long long的自动溢出
int T,n;bool flag;

struct String{
	int len,num;
	vector<ULL> hash;//记录hash值
	vector<int> word;//记录每个通配符的出现位置
	inline friend bool cmp1(const String a,const String b){return a.word[1]<b.word[1];}
	inline friend bool cmp2(const String a,const String b){return a.len-a.word[a.num]<b.len-b.word[b.num];}
	inline friend bool operator< (String a,String b){
		if(flag) return cmp1(a,b);
		else return cmp2(a,b);
	}
	inline void init(){
		len=0,num=0;
		hash.clear();hash.push_back(0);
		word.clear();word.push_back(0);//防止玄学错误,也是为了把数组下标处理为都+1
	}
	inline void build(string s){//计算hash值存下来
		for(string::iterator it=s.begin();it!=s.end();it++){
			hash.push_back(hash.back()*base+(*it));
			len++;
			if((*it)=='*'){
				num++;
				word.push_back(len);
			}
		}
	}
	inline ULL get_hash(int l,int r){return hash[r]-hash[l-1]*pow[r-l+1];}
	inline int get_suffix()const{return len-word[num];}
	inline bool match(String s){//有通配符匹配没有通配符
		int suf=get_suffix();//后缀长度
		if(word[1]-1+suf>s.len) return false;
		if(get_hash(1,word[1]-1)!=s.get_hash(1,word[1]-1)) return false;
		if(get_hash(word[num]+1,len)!=s.get_hash(s.len-suf+1,s.len)) return false;
		int l=word[1],r=s.len-suf;
		for(int i=1;i<num;i++){
			int length=word[i+1]-word[i]-1;
			ULL t=get_hash(word[i]+1,word[i+1]-1);
			while(1){
				if(l+length-1>r) return false;
				if(s.get_hash(l,l+length-1)==t){l+=length;break;}
				l++;
			}
		}
		return true;
	}
}s[N];

inline bool solve(){
	ULL hashval=0;int pos=0;
	for(int i=1;i<=n;i++){
		if(s[i].num) continue;//有通配符就跳过
		if(!pos){
			pos=i;
			hashval=s[i].hash[s[i].len];
		}
		else if(s[i].hash[s[i].len]!=hashval) return false;//没有通配符之间的
	}
	if(pos){
		for(int i=1;i<=n;i++)
			if(s[i].num&&(!s[i].match(s[pos]))) return false;//有通配符和没有通配符的
	}
	else{
		flag=true;sort(s+1,s+n+1);//按word[1]排序
		for(int i=1;i<n;i++){
			if(s[i].get_hash(1,s[i].word[1]-1)!=s[i+1].get_hash(1,s[i].word[1]-1)) //判断前缀
				return false;
		}
		flag=false;sort(s+1,s+n+1);//按word长度排序
		for(int i=1;i<n;i++){
			if(s[i].get_hash(s[i].word[s[i].num]+1,s[i].len)!=s[i+1].get_hash(s[i+1].len-s[i].get_suffix()+1,s[i+1].len)) //判断后缀
				return false;
		}
	}
	return true;
}

inline int read(){
	int x;cin>>x;
	return x;//...
}

int main(){
	ios::sync_with_stdio(false);//关同步cin都很快
	pow[0]=1;
	for(int i=1;i<M;i++)
		pow[i]=pow[i-1]*base;
	T=read();
	while(T--){
		cin>>n;
		for(int i=1;i<=n;i++){
			string ch;
			cin>>ch;
			s[i].init();
			s[i].build(ch);
		}
		puts(solve()?"Y":"N");
	}
}