Kattis - prefixfreecode—E - Prefix Free Code （字典树&&树状数组）

链接：

https://vjudge.net/contest/265795#problem/

题意：

给出 n个字符串，从中选出m个排列，问这m个字符构成的串在所有选出m个字符的排列中排第几。（类似康拓展开）

思路：

字符串排序，构造字典树并对每个单词标号，然后得到构成串的序列号，利用树状数组维护序列中某个位置的数后面有几个比他小的数。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;
const int maxn = 1e6 +1000;
struct Node
{
    bool judge;//是否是单词结尾点
    int next[26];
    int index; //单词编号，按字典序排序后的
} node[maxn];
int n,m,root,tot;
ll ans[maxn],c[maxn],sum;//ans:阶乘数，c：树状数组的c数组
int ID()
{
    node[tot].judge = false;
    memset(node[tot].next,-1,sizeof (node[tot].next));
    node[tot].index = 0;
    return tot++;
}
int lowbit(int x)
{
    return x&(-x);
}
ll getSum(int x)
{
    ll Sum = 0;
    for(int i = x; i; i-= lowbit(i))
    {
        Sum += c[i];
    }
    return Sum;
}
void addSum(int x,int val)
{
    for(int i = x + 1; i <= n; i += lowbit(i))//这里是从x+1开始更新的，感觉x也行
        c[i]+=val;
}
void BuildTree(int r,string s,int id)
{
    int len = s.size();
    for(int i = 0; i < len; i++)
    {
        int num = s[i] - 'a';
        if(node[r].next[num] == -1)
        {
            node[r].next[num] = ID();
        }
        r = node[r].next[num];
    }
    node[r].judge = true;
    node[r].index = id;
}
void init()//初始化操作
{
    tot = sum = 0;
    ans[m] = 1;
    ll res = (n - m + 1) % mod;
    for(int j = m - 1; j >= 1; j--)
    {
        ans[j] = (ans[j+1] * res) % mod;
        res = (res + 1) % mod;
    }
    memset(c,0,sizeof c);
    for(int i = 1 ; i <=n ; i++)
    {
        addSum(i,1);
    }
}

void searchTree(int r,string s)
{
    int len = s.size();
    int pos = 1;
    for(int  i = 0 ; i < len; i++)
    {
        int num = s[i] - 'a';
        r = node[r].next[num];
        if(node[r].judge)
        {
            int k = node[r].index;
            sum += (ans[pos] * getSum(k)) % mod;
            addSum(k,-1);
            pos++;
            r = root;
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    vector<string> v;
    string ss;
    cin>>n>>m;
    init();
    root = ID();
    for(int i = 0; i < n; i++)
    {
        cin>>ss;
        v.push_back(ss);
    }
    sort(v.begin(),v.end());
    for(int j = 0; j < v.size(); j++)
    {
        BuildTree(root,v[j],j+1);
    }
    cin>>ss;
    searchTree(root,ss);
    cout<<(sum+1)%mod<<endl;
    return 0;
}