【模拟】2017 Multi-University Training Contest 1 The Battle of Chibi

acm.hdu.edu.cn/showproblem.php?pid=5542

【Accepted】

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>

using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int maxn=1e5+2;
char str[maxn];
ll a[30][maxn];
bool vis[30];
int rk[30];
int l[30]; 
bool lower(int x,int y)
{
    if(l[x]!=l[y]) return l[x]<l[y];
    for(int i=maxn-1;i>=0;i--)
    {
        if(a[x][i]!=a[y][i]) return a[x][i]<a[y][i]; 
    }
    return false;
}
int n;
int main()
{
    int cas=0;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        memset(vis,false,sizeof(vis));
        memset(l,0,sizeof(l));
        for(int i=0;i<n;i++)
        {
            scanf("%s",str);
            int len=strlen(str);
            int cnt=0;
            for(int k=len-1;k>=0;k--)
            {
                a[str[k]-'a'][cnt++]++;
            }
            if(len!=1)
            {
                vis[str[0]-'a']=true; 
            }
            
        }
        for(int i=0;i<26;i++)
        {
            for(int k=0;k<maxn-1;k++)
            {
                a[i][k+1]+=(a[i][k]/26);
                a[i][k]=a[i][k]%26;
            }
            rk[i]=i;
        }
        for(int i=0;i<26;i++)
        {
            for(int k=maxn-1;k>=0;k--)
            {
                if(a[i][k]>0)
                {
                    l[i]=k;
                    break;
                }
            }
        }
        for(int i=0;i<26;i++)
        {
            for(int k=i+1;k<26;k++)
            {
                if(lower(rk[k],rk[i]))
                {
                    swap(rk[i],rk[k]);
                }
            }
        }
        for(int i=0;i<26;i++)
        {
            if(!vis[rk[i]])
            {
                int x=rk[i];
                for(int k=i-1;k>=0;k--)
                {
                    rk[k+1]=rk[k];
                }
                rk[0]=x;
                break;
            }
        }
        ll ans=0;
        for(int i=1;i<26;i++)
        {
            ll now=1;
            ll num=0;
            for(int k=0;k<=l[rk[i]];k++)
            {
                num=(num+(ll)a[rk[i]][k]*now%mod)%mod;
                now=(now*26)%mod;
            }
            ans=(ans+i*num%mod)%mod;
        }
        printf("Case #%d: %d
",++cas,(int)ans);
        
    }
    return 0;
 }