2020浙江省赛 - C. Crossword Validation（mp/字典树）

C. Crossword Validation

题意
给定一个小写字母和字符 ‘#’ 构成的 n*n 矩阵。给定一个字典，包含 m 个单词，每个单词有一个权值 wi。
可以从矩阵中得到若干个单词：按行遍历或者按列遍历，一段连续的不可扩展的小写字母串即为一个单词。

• 如果从矩阵中得到的单词在字典中不存在，那么输出 -1；
• 否则，输出所有得到单词的权值之和（重复单词可累加）。

思路1：mp标记
可以直接 mp 标记出字典中的所有单词，后面直接查询存不存在。
注：用 unordered_map。
Code

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

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define mem(a,b) memset(a,b,sizeof a)
#define int long long
#define fi first
#define se second
#define endl '\n'

/**/

const int N = 2010, mod = 1e9+7;
int T, n, m;
char a[N][N];

unordered_map<string,int> mp, f;

signed main(){
	Ios;
	cin >> T;
	while(T--)
	{
		cin >> n >> m;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cin >> a[i][j];
			}
		}
		
		mp.clear();
		
		for(int i=1;i<=m;i++)
		{
			string s;cin >> s;
			int x;cin>>x;
			mp[s] = x;
		}
		
		int ans = 0, flag = 1;
		for(int i=1;i<=n;i++)
		{
			string s;
			for(int j=1;j<=n;j++)
			{
				if(a[i][j] == '#'){
					if(s != "" && !mp.count(s)) flag = 0;
					else ans += mp[s];
					s = "";
				}
				else s += a[i][j];
			}
			if(s != "" && !mp.count(s)) flag = 0;
			else ans += mp[s];
		}
		
		for(int j=1;j<=n;j++)
		{
			string s;
			for(int i=1;i<=n;i++)
			{
				if(a[i][j] == '#'){
					if(s != "" && !mp.count(s)) flag = 0;
					else ans += mp[s];
					s = "";
				}
				else s += a[i][j];
			}
			if(s != "" && !mp.count(s)) flag = 0;
			else ans += mp[s];
		}
		
		if(!flag) cout << -1 << endl;
		else cout << ans << endl;
	}
	
	return 0;
}

思路2：字典树
先将所有字典中的单词插入到字典树中，后面查询的时候判断树中是否存在。
需要注意几个坑点：

• 插入字典树的所有单词长度总和不超过 4e6，所以直接开 tree[4e6, 26]，不需要用 mp；
• 空间限制卡得很紧，所以一定去掉 define 的 long long，还要注意最终答案开 longlong；
• 多组测试样例，需要初始化 cnt，tree 数组 和 is_end 数组，注意只要清空上组样例用过的位置（根据未初始化的 cnt）。

坑点比较多。

Code

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

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define ll long long //空间限制比较紧，把long long换掉！ 
#define endl '\n'

/**/

const int N = 1010, mod = 1e9+7;
int T, n, m;
char a[N][N];
int cnt;
int tree[4000010][30], is_end[4000010]; //注意到插入到字典树中的字符串长度总和不超过4e6，所以不需要开mp 

void insert(string s, int val)
{
	int root = 0;
	for(int i=0;i<s.size();i++)
	{
		int x = s[i] - 'a';
		if(!tree[root][x]) tree[root][x] = ++cnt;
		root = tree[root][x];
	}
	is_end[root] = val;
}

int find(string s)
{
	int root = 0;
	for(int i=0;i<s.size();i++)
	{
		int x = s[i] - 'a';
		if(!tree[root][x]) return 0;
		root = tree[root][x];
	}
	return is_end[root];
}

signed main(){
	Ios;
	cin >> T;
	while(T--)
	{
		cin >> n >> m;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cin >> a[i][j];
			}
		}
		
		//初始化时，只初始化上一个样例用过的位置 
		for(int i=0;i<=cnt;i++)
		{
			for(int j=0;j<26;j++)
				tree[i][j] = 0;
			is_end[i] = 0;
		}
		cnt = 0;
		
		for(int i=1;i<=m;i++)
		{
			string s;cin >> s;
			int x;cin>>x;
			insert(s, x);
		}
		
		ll ans = 0; //结果可能爆int！ 
		int flag = 1;
		for(int i=1;i<=n;i++)
		{
			string s;
			for(int j=1;j<=n;j++)
			{
				if(a[i][j] == '#'){
					int f = find(s);
					if(s != "" && !f) flag = 0;
					else ans += f;
					s = "";
				}
				else s += a[i][j];
			}
			int f = find(s);
			if(s != "" && !f) flag = 0;
			else ans += f;
		}
		
		for(int j=1;j<=n;j++)
		{
			string s;
			for(int i=1;i<=n;i++)
			{
				if(a[i][j] == '#'){
					int f = find(s);
					if(s != "" && !f) flag = 0;
					else ans += f;
					s = "";
				}
				else s += a[i][j];
			}
			int f = find(s);
			if(s != "" && !f) flag = 0;
			else ans += f;
		}
		
		if(!flag) cout << -1 << endl;
		else cout << ans << endl;
	}
	
	return 0;
}

---