Codeforces 965E 思路合并

参考  点击打开链接

可归结为一个合并过程。

先将字符串建成一颗trie树，然后对于每个节点我们结算 “所有在该节点子树中的字符串 的最小长度比该节点深度大的最优解集合”。如果当前节点不是某个字符串的终点，那么该节点深度记为一个新的最优值放到集合，并将原集合最大值删除，这样便，构成 所有在该节点子树中的字符串 的最小长度大于等于该节点深度的最优解集合。对于该节点时字符串终点时，只需要将之加入集合而不需要删除原集合最大值，因为又多了一个字符串。

注意这里的合并技巧，用大集合取吞噬小集合，这样实际运行很快，时间复杂度不会算，有会算大神的可以留言。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int maxn = 101010;
int end_[maxn],nex[maxn][30],cnt=0;
int Num(char s){return s - 'a';}
priority_queue<int> a[maxn];
int vec[maxn] , now = 0;
void insert(char* s, int dep,int f){
     if(s[dep] ==  '\0'){
           end_[f]  = 1;
           return ;
     }
     int id = Num(s[dep]);
     if(!nex[f][id]) nex[f][id] = ++cnt;
     //cout<<nex[f][id]<<" "<<s[dep]<<endl;
     insert(s , dep + 1 , nex[f][id]);
}
int n;
char str[(int)(1e5 + 10)];
int merge(int i ,int j){
    if(a[j].size() > a[i].size())
         swap(i , j);
    while(!a[j].empty()){
         int u = a[j].top();
         a[j].pop();
         a[i].push(u);
    }
    return i;
}
int dfs(int f,int dep){
    int hav = 0;
    for(int i = 0 ;i<26; i++){
        if(nex[f][i])
          dfs(nex[f][i],dep+1), hav = 1;
    }
    int pos = -1;
    for(int i =0 ; i<26;i++){
          if(nex[f][i]){
              if(pos == -1) pos = vec[nex[f][i]];
              else {
                  pos = merge(pos , vec[nex[f][i]]);
              }
          }
    }
    if(end_[f]) {
          if(!hav){
           vec[f] = ++now;
           a[vec[f]].push(dep);
          } else {
              vec[f] = pos;
              a[vec[f]].push(dep);
          }
    } else if(f > 0){
         vec[f] = pos;
         a[vec[f]].pop();
         a[vec[f]].push(dep);
    }
    return pos;
}
int main()
{
  scanf("%d",&n);
  gets(str);
  for(int i=1;i<=n;i++){
       gets(str);
       insert(str,0,0);
  }
  int pos = dfs(0,0);
  long long all = 0;
  while(!a[pos].empty()){
       int u = a[pos].top();
       a[pos].pop();
       all+=u;
  }
  cout<<all<<endl;
  return 0;
}