标签:
[UOJ#131][BZOJ4199][NOI2015]品酒大会
试题描述
一年一度的“幻影阁夏日品酒大会”隆重开幕了。大会包含品尝和趣味挑战两个环节,分别向优胜者颁发“首席品酒家”和“首席猎手”两个奖项,吸引了众多品酒师参加。
在大会的晚餐上,调酒师 Rainbow 调制了 n杯鸡尾酒。这 n杯鸡尾酒排成一行,其中第 i杯酒 (1≤i≤n ) 被贴上了一个标签 si ,每个标签都是 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)相似”的。特别地,对于任意的 1≤p,q≤n ,p≠q ,第 p 杯酒和第 q 杯酒都是“0相似”的。
在品尝环节上,品酒师 Freda 轻松地评定了每一杯酒的美味度,凭借其专业的水准和经验成功夺取了“首席品酒家”的称号,其中第 i 杯酒 (1≤i≤n ) 的美味度为 ai 。现在 Rainbow 公布了挑战环节的问题:本次大会调制的鸡尾酒有一个特点,如果把第 p 杯酒与第 q 杯酒调兑在一起,将得到一杯美味度为 apaq 的酒。现在请各位品酒师分别对于 r=0,1,2,…,n−1 ,统计出有多少种方法可以选出 2 杯“r相似”的酒,并回答选择 2 杯“r相似”的酒调兑可以得到的美味度的最大值。
输入
输入文件的第 1 行包含 1个正整数 n ,表示鸡尾酒的杯数。
第 2 行包含一个长度为 n 的字符串 S ,其中第 i个字符表示第 i 杯酒的标签。
第 3 行包含 n 个整数,相邻整数之间用单个空格隔开,其中第 i个整数表示第 i杯酒的美味度 ai 。
输出
输出文件包括 n行。第 i 行输出 2 个整数,中间用单个空格隔开。第 1个整数表示选出两杯“(i−1) 相似”的酒的方案数,第 2 个整数表示选出两杯“(i−1)相似”的酒调兑可以得到的最大美味度。若不存在两杯“(i−1)相似”的酒,这两个数均为0 。
输入示例1
10 ponoiiipoi 2 1 4 7 4 8 3 6 4 7
输出示例1
45 56 10 56 3 32 0 0 0 0 0 0 0 0 0 0 0 0 0 0
输入示例2
12 abaabaabaaba 1 -2 3 -4 5 -6 7 -8 9 -10 11 -12
输出示例2
66 120 34 120 15 55 12 40 9 27 7 16 5 7 3 -4 2 -4 1 -4 0 0 0 0
输入示例3
http://uoj.ac/download.php?type=problem&id=131
输出示例3
http://uoj.ac/download.php?type=problem&id=131
数据规模及约定
n ≤ 300000, |ai| ≤ 1000000000
题解
一看是串匹配的题,先进行后缀排序,计算LCP。
不难发现“r相似”,随着r的减小,方案数和最大美味度一定不降,所以不妨从n-1到0枚举r,再进行累加。
两个后缀“r相似”当且仅当height数组中两个后缀之间最小值大于等于r,考虑使用并查集,将排好序的每个后缀看成一个节点,两个相邻后缀之间的匹配看成一条边,于是当这个匹配值大于等于r时,把这两个相邻的后缀所在的连通块合并成一个。
需要维护连通块的大小(siz),连通块中的最小美味度(minv)和最大美味度(maxv)。
#include <iostream> #include <cstdio> #include <algorithm> #include <cmath> #include <stack> #include <vector> #include <queue> #include <cstring> #include <string> #include <map> #include <set> using namespace std; const int BufferSize = 1 << 16; char buffer[BufferSize], *Head, *tail; inline char Getchar() { if(Head == tail) { int l = fread(buffer, 1, BufferSize, stdin); tail = (Head = buffer) + l; } return *Head++; } int read() { int x = 0, f = 1; char c = Getchar(); while(!isdigit(c)){ if(c == ‘-‘) f = -1; c = Getchar(); } while(isdigit(c)){ x = x * 10 + c - ‘0‘; c = Getchar(); } return x * f; } #define maxn 300010 #define oo 1ll << 60 #define LL long long int n, val[maxn]; char S[maxn]; int m, rank[maxn], height[maxn], Ws[maxn], sa[maxn]; bool cmp(int *a, int p1, int p2, int len) { return a[p1] == a[p2] && a[p1+len] == a[p2+len]; } void ssort() { int *x = rank, *y = height; m = 0; for(int i = 1; i <= n; i++) Ws[i] = 0; for(int i = 1; i <= n; i++) Ws[x[i] = S[i]]++, m = max(m, (int)S[i]); for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1]; for(int i = n; i; i--) sa[Ws[x[i]]--] = i; for(int pos = 0, j = 1; pos < n; j <<= 1, m = pos) { pos = 0; for(int i = n - j + 1; i <= n; i++) y[++pos] = i; for(int i = 1; i <= n; i++) if(sa[i] > j) y[++pos] = sa[i] - j; for(int i = 1; i <= m; i++) Ws[i] = 0; for(int i = 1; i <= n; i++) Ws[x[y[i]]]++; for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1]; for(int i = n; i; i--) sa[Ws[x[y[i]]]--] = y[i]; swap(x, y); pos = 1; x[sa[1]] = 1; for(int i = 2; i <= n; i++) x[sa[i]] = cmp(y, sa[i], sa[i-1], j) ? pos : ++pos; } return ; } void calch() { for(int i = 1; i <= n; i++) rank[sa[i]] = i; for(int i = 1, j, k = 0; i <= n; height[rank[i++]] = k) for(k ? k-- : 0, j = sa[rank[i]-1]; S[i+k] == S[j+k]; k++) ; return ; } int fa[maxn], siz[maxn], minv[maxn], maxv[maxn]; int findset(int x) { return x == fa[x] ? x : fa[x] = findset(fa[x]); } int pos[maxn], cntv[maxn]; LL ansc[maxn], ansm[maxn]; LL max(LL a, LL b, LL c, LL x, LL y, LL z) { return max(max(max(a * b, b * c), a * c), max(max(x * y, y * z), x * z)); } int main() { read(); n = 0; char tc = Getchar(); while(!isalpha(tc)) tc = Getchar(); while(isalpha(tc)) S[++n] = tc, tc = Getchar(); for(int i = 1; i <= n; i++) val[i] = read(); ssort(); calch(); for(int i = 2; i <= n; i++) cntv[height[i]]++, m = max(m, height[i]); for(int i = m; i >= 0; i--) cntv[i] += cntv[i+1]; for(int i = n; i > 1; i--) pos[cntv[height[i]]--] = i; for(int i = 1; i <= n; i++) fa[i] = i, siz[i] = 1, minv[i] = maxv[i] = val[sa[i]]; for(int i = 0; i <= n; i++) ansm[i] = -oo; // for(int i = 1; i <= n; i++) printf("%d ", height[i]); putchar(‘\n‘); for(int i = 1; i <= n; i++) if(pos[i] > 1) { // printf("%d\n", pos[i]); int u = findset(pos[i]-1), v = findset(pos[i]); ansc[height[pos[i]]] += (LL)siz[u] * siz[v]; ansm[height[pos[i]]] = max(ansm[height[pos[i]]], max((LL)minv[u] * minv[v], (LL)maxv[u] * maxv[v])); siz[u] += siz[v]; minv[u] = min(minv[u], minv[v]); maxv[u] = max(maxv[u], maxv[v]); siz[v] = minv[v] = maxv[v] = 0; fa[v] = u; } for(int i = n - 1; i >= 0; i--) ansc[i] += ansc[i+1], ansm[i] = max(ansm[i], ansm[i+1]); for(int i = 0; i < n; i++) { if(!ansc[i]) ansm[i] = 0; printf("%lld %lld\n", ansc[i], ansm[i]); } return 0; }
[UOJ#131][BZOJ4199][NOI2015]品酒大会 后缀数组 + 并查集
标签:
原文地址:http://www.cnblogs.com/xiao-ju-ruo-xjr/p/5400947.html