pac style noi2015 i++ 特點 += -s scanf oid

BZOJ_4199_[Noi2015]品酒大會_後綴自動機

Description

一年一度的“幻影閣夏日品酒大會”隆重開幕了。大會包含品嘗和趣味挑戰兩個環節，分別向優勝者頒發“首席品

酒家”和“首席獵手”兩個獎項，吸引了眾多品酒師參加。在大會的晚餐上，調酒師Rainbow調制了 n 杯雞尾酒。 這 n 杯雞尾酒排成一行，其中第 i 杯酒 (1≤i≤n) 被貼上了一個標簽 s_i ，每個標簽都是 26 個小寫英文字母 之一。設 Str(l,r) 表示第 l 杯酒到第 r 杯酒的 r-l+1 個標簽順次連接構成的字符串。若 Str(p,po)=Str(q,qo ) ，其中 1≤p≤po≤n,1≤q≤qo≤n,p≠q,po-p+1=qo-q+1=r ，則稱第 p 杯酒與第 q 杯酒是“ r 相似”的。當 然兩杯“ r 相似”(r>1)的酒同時也是“ 1 相似”、“ 2 相似”、……、“ (r-1) 相似”的。在品嘗環節上， 品酒師Freda輕松地評定了每一杯酒的美味度，憑借其專業的水準和經驗成功奪取了“首席品酒家”的稱號，其中 第 i 杯酒 (1≤i≤n) 的美味度為 a_i 。現在Rainbow公布了挑戰環節的問題：本次大會調制的雞尾酒有一個特點 ，如果把第 p 杯酒與第 q 杯酒調兌在一起，將得到一杯美味度為 a_p a_q 的酒。現在請各位品酒師分別對於 r= 0,1,2,?,n-1 ，統計出有多少種方法可以選出 2 杯“ r 相似”的酒，並回答選擇 2 杯“ r 相似”的酒調兌可以 得到的美味度的最大值。

Input

輸入文件的第1行包含1個正整數 n ，表示雞尾酒的杯數。 第 2 行包含一個長度為 n 的字符串 S ，其中第 i 個字符表示第 i 杯酒的標簽。 第 3 行包含 n 個整數，相鄰整數之間用單個空格隔開，其中第 i 個整數表示第 i 杯酒的美味度 a_i 。 n=300,000 |a_i |≤1,000,000,000

Output

輸出文件包括 n 行。第 i 行輸出 2 個整數，中間用單個空格隔開。

第 1 個整數表示選出兩杯“ (i-1)" " 相似”的酒的方案數， 第 2 個整數表示選出兩杯“ (i-1) 相似”的酒調兌可以得到的最大美味度。 若不存在兩杯“ (i-1) 相似”的酒，這兩個數均為 0 。

Sample Input

10
ponoiiipoi
2 1 4 7 4 8 3 6 4 7

Sample Output

45 56
10 56
3 32
0 0
0 0
0 0
0 0
0 0
0 0
0 0

---

先只考慮相似度等於i的兩個答案ans1[i],ans2[i]，最後再整個遞推一遍。

ans1[i],ans2[i]對應後綴樹上深度為i的答案。

ans1[i]是比較好求的，類似差異那道題，用siz[fa[p]]*siz[p]計算貢獻，因為後綴樹上LCA深度就等於兩子串的LCP長度。

然後ans2[i],由於題中a[i]可能有負數，要同時維護maxn[i],minn[i]表示i子樹內的最大/最小值。

代碼：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 300050
typedef long long ll;
#define inf 0x3f3f3f3f
int n,ch[N<<1][26],fa[N<<1],dep[N<<1],cnt=1,lst=1,val[N];
int ws[N<<1],a[N<<1],siz[N<<1],maxn[N<<1],minn[N<<1];
char w[N];
ll ans1[N],ans2[N];
int head[N<<1],to[N<<1],nxt[N<<1],tot;
inline void add(int u,int v) {
	to[++tot]=v; nxt[tot]=head[u]; head[u]=tot;
}
void insert(int x) {
	int p=lst,np=++cnt,q,nq;
	lst=np; dep[np]=dep[p]+1;
	for(;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
	if(!p) fa[np]=1;
	else {
		q=ch[p][x];
		if(dep[q]==dep[p]+1) fa[np]=q;
		else {
			fa[nq=++cnt]=fa[q];
			dep[nq]=dep[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[np]=fa[q]=nq;
			for(;p&&ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
		}
	}
}
void dfs(int x) {
	int i,d=dep[x];
	for(i=head[x];i;i=nxt[i]) {
		int p=to[i];
		dfs(p);
		ans1[d]+=1ll*siz[x]*siz[p];
		if(minn[x]<inf&&minn[p]<inf) ans2[d]=max(ans2[d],max(1ll*maxn[x]*maxn[p],1ll*minn[x]*minn[p]));
		maxn[x]=max(maxn[x],maxn[p]);
		minn[x]=min(minn[x],minn[p]);
		siz[x]+=siz[to[i]];
	}
}
int main() {
	scanf("%d%s",&n,w+1);
	int i;
	for(i=1;i<=n;i++) scanf("%d",&val[i]);
	memset(minn,0x3f,sizeof(minn)); memset(maxn,0x80,sizeof(maxn));
	memset(ans2,0x80,sizeof(ans2));
	for(i=n;i;i--) insert(w[i]-‘a‘),minn[lst]=maxn[lst]=val[i],siz[lst]++;
	for(i=2;i<=cnt;i++) add(fa[i],i);
	dfs(1);
	for(i=n-2;i>=0;i--) ans1[i]+=ans1[i+1],ans2[i]=max(ans2[i],ans2[i+1]);
	for(i=0;i<n;i++) {
		if(ans1[i]) printf("%lld %lld\n",ans1[i],ans2[i]);
		else puts("0 0");
	}
}

BZOJ_4199_[Noi2015]品酒大會_後綴自動機