Good Article Good sentence HDOJ4416

做了这题,不得不说对后缀数组的理解还是不深刻(感觉这题SAM也可以搞,等学透了SAM再来看看这题),题意就是求第一个串S1中有多少个子串不是其他串Si(i > 1)的子串,老规矩,先把串都连接起来,中间用不同的字符连接起来, 然后求每个S1的后缀与其他串Si的后缀的lcp,这表明该后缀有lcp的子串是其他串的子串,这里求lcp不用把每个S1的后缀与Si的后缀求lcp,根据后缀数组的性质,只要从左右两边扫描一遍即可(具体见代码),最后排除掉S1串中本身重复的子串, 还要注意的是此题有n=0的时候, WA到吐。


#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <queue>
#include <algorithm>
#include <vector>
#include <cstring>
#include <stack>
#include <cctype>
#include <utility>   
#include <map>
#include <string>  
#include <climits> 
#include <set>
#include <string>    
#include <sstream>
#include <utility>   
#include <ctime>
#include <bitset>

using std::priority_queue;
using std::vector;
using std::swap;
using std::stack;
using std::sort;
using std::max;
using std::min;
using std::pair;
using std::map;
using std::string;
using std::cin;
using std::cout;
using std::set;
using std::queue;
using std::string;
using std::stringstream;
using std::make_pair;
using std::getline;
using std::greater;
using std::endl;
using std::multimap;
using std::deque;
using std::unique;
using std::lower_bound;
using std::random_shuffle;
using std::bitset;

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PAIR;
typedef multimap<int, int> MMAP;
typedef LL TY;

const int MAXN(300010);
const int MAXM(50010);
const int MAXE(100010);
const int MAXK(6);
const int HSIZE(131313);
const int SIGMA_SIZE(52);
const int MAXH(19);
const int INFI((INT_MAX-1) >> 1);
const ULL BASE(31);
const LL LIM(10000000);
const int INV(-10000);
const int MOD(1000000007);
const double EPS(1e-7);

template<typename T> void checkmax(T &a, T b){if(b > a) a = b;}
template<typename T> void checkmin(T &a, T b){if(b < a) a = b;}
template<typename T> T ABS(const T &a){return a < 0? -a: a;}

char str[100010];

struct SA
{
	int S[MAXN];
	int sa[MAXN], t1[MAXN], t2[MAXN], cnt[MAXN], len, M;
	void init(int tl, int tm = 128)
	{
		len = tl;
		M = tm;
		int *p1 = t1;
		int *p2 = t2;
		for(int i = 0; i < M; ++i) cnt[i] = 0;
		for(int i = 0; i <= len; ++i) ++cnt[p1[i] = S[i]];
		for(int i = 1; i < M; ++i) cnt[i] += cnt[i-1];
		for(int i = len; i >= 0; --i) sa[--cnt[p1[i]]] = i;
		int temp = 1;
		for(int k = 1; temp <= len; k <<= 1)
		{
			temp = 0;
			for(int i = len-k+1; i <= len; ++i) p2[temp++] = i;
			for(int i = 0; i <= len; ++i)
				if(sa[i] >= k)
					p2[temp++] = sa[i]-k;
			for(int i = 0; i < M; ++i) cnt[i] = 0;
			for(int i = 0; i <= len; ++i) ++cnt[p1[p2[i]]];
			for(int i = 1; i < M; ++i) cnt[i] += cnt[i-1];
			for(int i = len; i >= 0; --i) sa[--cnt[p1[p2[i]]]] = p2[i];
			swap(p1, p2);
			temp = 1;
			p1[sa[0]] = 0;
			for(int i = 1; i <= len; ++i)
				p1[sa[i]] = p2[sa[i-1]] == p2[sa[i]] && p2[sa[i-1]+k] == p2[sa[i]+k]? temp-1: temp++;
			M = temp;
		}
	}
	int rank[MAXN], height[MAXN];
	void getHeight()
	{
		int k = 0;
		for(int i = 0; i <= len; ++i)
			rank[sa[i]] = i;
		for(int i = 0; i < len; ++i)
		{
			if(k) --k;
			int j = sa[rank[i]-1];
			while(S[i+k] == S[j+k]) ++k;
			height[rank[i]] = k;
		}
	}
	/*
	int Log[MAXN];
	int table[MAXH][MAXN];
	void initLog()
	{
		Log[0] = -1;
		for(int i = 1; i < MAXN; ++i)
			Log[i] = (i&(i-1))?Log[i-1]: Log[i-1]+1;
	}
	void initRMQ()
	{
		for(int i = 1; i <= len; ++i)
			table[0][i] = height[i];
		for(int i = 1; (1 << i) <= len; ++i)
			for(int j = 1; j+(1 << i)-1 <= len; ++j)
				table[i][j] = min(table[i-1][j], table[i-1][j+(1 << (i-1))]);
	}
	int lcp(int a, int b)
	{
		a = rank[a];
		b = rank[b];
		if(a > b) swap(a, b);
		++a;
		int temp = Log[b-a+1];
		return min(table[temp][a], table[temp][b-(1 << temp)+1]);
	}
	*/
}sa;

int len1, len2;
int rec[MAXN];

int main()
{
	int TC, n_case(0);
	scanf("%d", &TC);
	while(TC--)
	{
		int n;
		scanf("%d", &n);
		scanf("%s", str);
		len2 = len1 = strlen(str);
		for(int i = 0; i <= len1; ++i)
			sa.S[i] = str[i];
		int temp = 'z'+1;
		for(int i = 0; i < n; ++i)
		{
			scanf("%s", str);
			sa.S[len2++] = temp++;
			int tlen = strlen(str);
			for(int j = 0; j <= tlen; ++j)
				sa.S[len2+j] = str[j];
			len2 += tlen;
		}
		sa.init(len2, temp);
		sa.getHeight();
		printf("Case %d: ", ++n_case);
		LL ans;
		ans = (LL)len1*(len1+1)/2;
		if(n == 0)
		{
			for(int i = 1; i <= len1; ++i)
				ans -= sa.height[i];
			printf("%I64d\n", ans);
			continue;
		}
		temp = INFI;
		for(int i = 1; i <= len2; ++i)
			if(sa.sa[i] <= len1)
			{
				checkmin(temp, sa.height[i]);
				rec[i] = temp;
			}
			else
				temp = INFI;
		temp = INFI;
		for(int i = len2; i >= 1; --i)
			if(sa.sa[i-1] <= len1)
			{
				checkmin(temp, sa.height[i]);
				checkmax(rec[i-1], temp);
			}
			else
				temp = INFI;
		for(int i = 0; i < len1; ++i)
			if(sa.sa[sa.rank[i]-1] <= len1)
				checkmax(rec[sa.rank[i]], sa.height[sa.rank[i]]);
		for(int i = 0; i < len1; ++i)
			ans -= rec[sa.rank[i]];
		printf("%I64d\n", ans);
	}
	return 0;
}




你可能感兴趣的:(Good Article Good sentence HDOJ4416)