[Codeforces Round #373 DIV1E (CF718E)] Matvey's Birthday

题意

给定一个字符串，字符集大小为8
将每个位置当做一个点，相邻位置有长度为1的连边，任意相同字母的位置之间有长度为1的连边。求图的直径和直径的长度。
$n\le 10^5$

题解

求图的直径需要求任意两点间的最短路，这是不现实的。

设$DIS(i,j)$为位置$i$与位置$j$之间的最短路，那么

$$DIS(i,j) = min\{|i-j|,min\{dis(i,c)+1+dis(j,c)\}\}$$
其中 $dis(i,c)$表示从某个位置到达字符 $x$的最短距离。

为什么$dis(i,c)+1+dis(j,c)$里面要 $+1$呢？因为如果 $i,j$都到达了同一个 $c$，那么这个长度一定大于等于$|i-j|$。

同时我们能得到这样的结论，树的直径小于字符集大小的二倍，即15，这个串就是一个例子aabbccddeeffgghh

枚举$i$，同时枚举$j$

1. $|i-j|<16$，此时直接由$DIS(i,j) = min\{|i-j|,min\{dis(i,c)+1+dis(j,c)\}\}$求出
2. $|i-j|\ge 16$，不能直接枚举$j$了，此时$DIS(i,j)=min\{dis(i,c)+1+dis(j,c)\}$

现在设法求出$|i-j|\ge 16$时的$max\{DIS(i,j)\}$，注意此时的$i$是固定的，$j$是不确定的。

再设$d(c_1,c_2)$代表字符$c_1$与$c_2$之间的最短距离。那么$d(s_j,c)\le dis(j,c)\le d(s_j,c)+1$，也就是要么$s_j$与$c$之间最短路恰好是$j$位置到$c$的最短路，要么需要再跳一步。

预处理求出$dis$数组与$d$数组，再设一个$mask$数组，$mask(j)$对于每一个位置$j$，用二进制存下$dis(j,c)-d(s_j,c), 0\le c < 8$
再设数组$cnt(x,mask)$来统计$j (|i-j|\ge16) , s_j=x 且 mask(j) = mask$的数量。

$i$确定时先确定$s_j$再枚举$mask$，此时

$$min\{dis(i,c)+1+dis(j,c)\} =min\{ dis(i,c)+1+d(s_j,c)+mask_{c}\}$$
这样就能求出 $i,s_j,mask$确定时的最短路并更新答案。
时间复杂度 $O(n\cdot |C|^3+n\cdot |C|^2\cdot 2^{|C|})$，其中 $|C|$为字符集大小。

代码

/// by ztx
/// blog.csdn.net/hzoi_ztx

#define Rep(i,l,r) for(i=(l);i<=(r);i++)
#define rep(i,l,r) for(i=(l);i< (r);i++)
#define Rev(i,r,l) for(i=(r);i>=(l);i--)
#define rev(i,r,l) for(i=(r);i> (l);i--)
#define Each(i,v)  for(i=v.begin();i!=v.end();i++)
#define r(x)   read(x)
typedef long long ll ;
typedef double lf ;
int CH , NEG ;
template <typename TP>inline void read(TP& ret) {
    ret = NEG = 0 ; while (CH=getchar() , CH<'!') ;
    if (CH == '-') NEG = true , CH = getchar() ;
    while (ret = ret*10+CH-'0' , CH=getchar() , CH>'!') ;
    if (NEG) ret = -ret ;
}

#define  maxn  100010LL
#define  infi  0x3f3f3f3fLL

#define  id(c) (c+n+1)
int n, ans;
ll cnt;
int dis[maxn][8], d[8][8], mask[maxn],c[8][256];
char s[maxn];
std::queue<int> q;
inline void bfs(int x) {
    int i, u, v;
    dis[id(x)][x] = 0;
    Rep (i,1,n) if (s[i] == x)
        dis[i][x] = 0, q.push(i);
    while (!q.empty())
        if (u = q.front(), q.pop(), u <= n) {
            if (v = u+1, 1<=v && v<=n && dis[v][x]==infi) {
                dis[v][x] = dis[u][x]+1, q.push(v);
                if (dis[id(s[v])][x] == infi)
                    dis[id(s[v])][x] = dis[u][x]+1, q.push(id(s[v]));
            }
            if (v = u-1, 1<=v && v<=n && dis[v][x]==infi) {
                dis[v][x] = dis[u][x]+1, q.push(v);
                if (dis[id(s[v])][x] == infi)
                    dis[id(s[v])][x] = dis[u][x]+1, q.push(id(s[v]));
            }
        }
        else Rep (i,1,n) if (id(s[i])==u && dis[i][x]==infi)
            dis[i][x] = dis[u][x]+1, q.push(i);
}

int main() {
    int i, j, k, l, now, x;
    r(n), scanf("%s", s+1);
    Rep (i,1,n) s[i] -= 'a';
    memset(dis, 0x3f, sizeof dis);
    while (!q.empty()) q.pop();
    rep (i,0,8) {
        bfs(i);
        rep (x,0,8) d[x][i] = dis[id(x)][i];
    }
    Rep (i,1,n) rep (x,0,8)
        if (dis[i][x] > d[s[i]][x]) mask[i] |= 1<<x;
    Rep (i,1,n) {
        rep (j,std::max(i-15,1),i) {
            now = i-j;
            rep (k,0,8) now = std::min(now, dis[j][k]+dis[i][k]+1);
            if (now == ans) cnt ++ ;
            if (now > ans) ans = now, cnt = 1;
        }
        int t = i-16;
        if (t >= 1) c[s[t]][mask[t]] ++ ;
        rep (x,0,8) rep (k,0,256)
            if (c[x][k]) {
                now = infi;
                rep (l,0,8) now = std::min(now, d[x][l]+1+dis[i][l]+(k&(1<<l)));
                if (now == ans) cnt += c[x][k];
                if (now > ans) ans = now, cnt = c[x][k];
            }
    }
    printf("%d %lld\n", ans, cnt);
    END: getchar(), getchar();
    return 0;
}