2982. 找出出现至少三次的最长特殊子字符串 II

字典树思路

2982. 找出出现至少三次的最长特殊子字符串 II_第1张图片

用字典树搞一下就好了,比如aaaaa :   a存5次 aa 4次以此类推~

字典树板子复习:P8306 【模板】字典树

这里这个清空方式 很好 因为很多时候memset   T

#include
#include
using namespace std;


const int N = 3e6+10;
int n,m;
string s;
int tire[N][70],cnt[N],idx;


void insert(){
	int p = 0;
	for(auto &t:s){
		int u;
		if(t>='a'&&t<='z')u = t-'a';
		else if(t>='A'&&t<='Z')u = t-'A'+ 27;
		else u = t -'0'+54;
		if(!tire[p][u]) tire[p][u]=++idx;
		p = tire[p][u];	
		cnt[p]++;
	}
}

int query(){
	int p=0;
	for(auto &t:s){
		int u;
		if(t>='a'&&t<='z')u = t-'a';
		else if(t>='A'&&t<='Z')u = t-'A'+ 27;
		else u = t -'0'+54;
		if(!tire[p][u]) return 0;
		p = tire[p][u];
	}
	
	return cnt[p];
}





void solve()
{
	cin>>n>>m;
	
	for(int i=0;i<=idx;i++)
	 for(int j=0;j<=65;j++)
	  tire[i][j] = 0;
	  
	for(int i=0;i<=idx;i++)cnt[i] = 0;
	idx = 0;
	while(n--)
	{
		cin>>s;insert();
	}
	
	while(m--){
		cin>>s;cout<>_;
	while(_--)solve();
}


本题代码;

const int N = 5e5+10;
int tr[N][27],cnt[N],idx;
class Solution {
public:

    string str;
    void insert(){
    	int ts = str.size();
        int p = 0;
        for(auto &t:str){
            int u = t - 'a';
            if(!tr[p][u])tr[p][u] = ++idx;
            p = tr[p][u];
            cnt[p]+=ts--;
        }

    }

    int find(char c){ 
        int p = 0;
        int t = 0;
        int u = c-'a';
        while(cnt[tr[p][u]]>=3){
        	t++;
        	p = tr[p][u];
        }
        return t;
    }

    int maximumLength(string s) {
        for(int i=0;i<=idx;i++)
         for(int j=0;j<=26;j++)
          tr[i][j] = 0;
        for(int i=0;i<=idx;i++)cnt[i] = 0;
        idx = 0;
        char c = s[0];
        int t = 1;
        for(int i=1;i

你可能感兴趣的:(数据结构,算法,数据结构,c++,leetcode,学习)