題目鏈接：點擊查看

題目大意：給出n個互不相同的字符串，現(xiàn)在要求n個互不相同的前綴，問所有前綴的長度和最短是多少

題目分析：因為涉及到了前綴，所以我們可以用字典樹來解決，又因為需要長度和最短，所以我們可以貪心來做，具體實現(xiàn)還是需要借助stl里的優(yōu)先隊列或者set容器，時間復(fù)雜度都是nlogn，這里我是用優(yōu)先隊列做的，在建立字典樹的時候，對于結(jié)尾的字母，我們需要標(biāo)記一下，最后我們可以得到一棵每個節(jié)點非0即1的字典樹（因為題目保證了兩兩字符串互不相同，所以肯定不可能有互相干擾的節(jié)點），然后這棵樹中肯定有n個1，所以我們盡可能的讓這n個1的深度在樹上變小即可，我們可以在每個節(jié)點都維護一個優(yōu)先隊列，然后根據(jù)樹的遍歷，利用dfs深搜，最初到達(dá)字典樹的最深一層，也就是葉子節(jié)點開始搜索：

若當(dāng)前位置的節(jié)點是1的話，直接將其深度加入當(dāng)前優(yōu)先隊列

若當(dāng)前位置的節(jié)點是0的話，就判斷一下當(dāng)前優(yōu)先隊列的最大值與當(dāng)前深度的關(guān)系

若深度小于優(yōu)先隊列的最大值，就將之前儲存的最大值的那個1放到這個0的位置

若深度大于優(yōu)先隊列的最大值，則忽略

如此往復(fù)的回溯，到最后我們字典樹的根節(jié)點的優(yōu)先隊列，肯定是有n個元素的，也就是本題的答案了，記錄一下就好了

代碼：

#include<iostream> #include<cstdlib> #include<string> #include<cstring> #include<cstdio> #include<algorithm> #include<climits> #include<cmath> #include<cctype> #include<stack> #include<queue> #include<list> #include<vector> #include<set> #include<map> #include<sstream> #include<unordered_map> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=1e5+100;priority_queue<int>q[N];char s[N];int cnt=0;int trie[N][26];bool vis[N];void insert() {int pos=0;for(int i=0;s[i];i++){int to=s[i]-'a';if(!trie[pos][to])trie[pos][to]=++cnt;pos=trie[pos][to];}vis[pos]=true; }void merge(int a,int b) {while(q[b].size()){q[a].push(q[b].top());q[b].pop(); } }void dfs(int pos,int deep) {for(int i=0;i<26;i++)if(trie[pos][i]){dfs(trie[pos][i],deep+1);merge(pos,trie[pos][i]);}if(!deep)return;if(vis[pos]){q[pos].push(deep);}else{if(q[pos].top()>deep){q[pos].pop();q[pos].push(deep);}} }int main() { // freopen("input.txt","r",stdin); // ios::sync_with_stdio(false);int n;scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%s",s);insert();}dfs(0,0);int ans=0;while(q[0].size()){ans+=q[0].top();q[0].pop();}printf("%d\n",ans);return 0; }

?

總結(jié)

以上是生活随笔為你收集整理的CodeForces - 965E Short Code (字典树上贪心)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。