首页 > 关于5.11字节跳动笔试第二题。
头像
奥利给给给啊啊
编辑于 2020-05-12 17:41
+ 关注

关于5.11字节跳动笔试第二题。

第二题和leetcode139很像的链接为:https://leetcode-cn.com/problems/word-break/
注意到第二题的数据范围为字符串的长度<=50000,如果单纯使用力扣代码,此时的复杂度为o(n^2),可以达到10^10级别,那么肯定无法AC。
那么考虑到给定的一个条件:可划分的字符串长度<=20。可以考虑去做一步优化,从当前位置的前20开始查起,然后去做。优化代码为:
#include<bits/stdc++.h>
using namespace std;
int main(){
	const int num=1e9+7;
	unordered_map<string,bool>m; 
	string str;
	cin>>str;
	int N;
	cin>>N;
	while(N--){
		string tmp;
		cin>>tmp;
		m[tmp]=true;
	}
	int n=str.size();
	vector<int>dp(n+1,0);
	dp[0]=1;
	for(int i=0;i<n;i++){
		for(int j=max(0,i-19);j<=i;j++){
			string tmp=str.substr(j,i-j+1);
			if(m[tmp]){
				dp[i+1]=(dp[i+1])%num+(dp[j]%num);
			}
		}
	}
	cout<<dp[n]<<endl;
	return 0;
} 
这样的话可以过0.7,依旧无法AC。
但是考虑到状态转移中,当前状态最多与前20个状态有关系,我们可以对状态做一步缩小化:
#include<bits/stdc++.h>
using namespace std;
int main(){
	const int num=1e9+7;
	unordered_map<string,bool>m; 
	string str;
	cin>>str;
	int N;
	cin>>N;
	while(N--){
		string tmp;
		cin>>tmp;
		m[tmp]=true;
	}
	int n=str.size();
	vector<int>dp(21,0);
	dp[0]=1;
	for(int i=0;i<n;i++){
		for(int j=max(0,i-19);j<=i;j++){
			string tmp=str.substr(j,i-j+1);
			if(m[tmp]){
				int t1=((i+1))%20;
				int t2=(j%20);
				dp[t1]=(dp[t1])%num+(dp[t2]%num);
			}
		}
	}
	cout<<dp[n%20]<<endl;
	return 0;
} 

这样整体来看的化,空间缩小了,也许能AC。但是考试的时候自己是按照第一份代码做的,所以只过了0.7。
当然这些只是自己的一些看法,欢迎大家讨论。

全部评论

(2) 回帖
加载中...
话题 回帖

推荐话题

相关热帖

历年真题 真题热练榜 24小时
技术(软件)/信息技术类
查看全部

近期精华帖

热门推荐