SDOI 2016 Round1 Day2

生成魔咒

/*
后缀数组+双向链表
参照：https://blog.csdn.net/clove_unique/article/details/53911757
*/
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+;
int n,cnt,c[N],ans[N],s[N],t[N],rank[N],trank[N],sa[N],tsa[N],h[N],nxt[N],pre[N];
long long res;
void DA(){
    int p,maxx=cnt;
    memset(c,,sizeof c);
    for(int i=;i<=n;i++) c[rank[i]=s[i]]++;
    for(int i=;i<=maxx;i++) c[i]+=c[i-];
    for(int i=n;i;i--) sa[c[rank[i]]--]=i;
    trank[sa[]]=p=;
    for(int i=;i<=n;i++){
        if(rank[sa[i]]!=rank[sa[i-]]) p++;
        trank[sa[i]]=p;
    }
    for(int i=;i<=n;i++) rank[i]=trank[i];
    for(int k=;p<n;k<<=,maxx=p){
        p=;
        for(int i=n-k+;i<=n;i++) tsa[++p]=i;
        for(int i=;i<=n;i++) if(sa[i]>k) tsa[++p]=sa[i]-k;
        memset(c,,sizeof c);
        for(int i=;i<=n;i++) trank[i]=rank[tsa[i]];
        for(int i=;i<=n;i++) c[trank[i]]++;
        for(int i=;i<=maxx;i++) c[i]+=c[i-];
        for(int i=n;i;i--) sa[c[trank[i]]--]=tsa[i];
        trank[sa[]]=p=;
        for(int i=;i<=n;i++){
            if(rank[sa[i]]!=rank[sa[i-]]||rank[sa[i]+k]!=rank[sa[i-]+k]) p++;
            trank[sa[i]]=p;
        }
        for(int i=;i<=n;i++) rank[i]=trank[i];
    }
    for(int i=,k=;i<=n;i++){
        int j=sa[rank[i]-];
        while(s[i+k]==s[j+k]) k++;
        h[rank[i]]=k;if(k>) k--;
    }
}
int main(){
    scanf("%d",&n);
    for(int i=;i<=n;i++) scanf("%d",&s[i]),t[i]=s[i];
    for(int i=;i<=n/;i++) swap(s[i],s[n-i+]);
    sort(t+,t+n+);
    cnt=unique(t+,t+n+)-(t+);
    for(int i=;i<=n;i++) s[i]=lower_bound(t+,t+cnt+,s[i])-t;
    DA();
    for(int i=;i<=n;i++){
        nxt[i]=i+;
        pre[i]=i-;
    }
    for(int i=;i<=n;i++){
        int k=rank[i];
        ans[i]=n-i+-max(h[k],h[nxt[k]]);
        h[nxt[k]]=min(h[k],h[nxt[k]]);
        nxt[pre[k]]=nxt[k];
        pre[nxt[k]]=pre[k];
    }
    for(int i=n;i;i--){
        res+=ans[i];
        printf("%lld\n",res);
    }
    return ;
}

排列计数

/*
ans=C(n,m)*dp(n-m);|dp(i)表示i的错排数
*/
#include<cstdio>
using namespace std;
typedef long long ll;
inline ll read(){
    ll x=;char ch=getchar();
    while(ch<''||ch>''){ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x;
}

#define FRE(name) freopen(#name".in","r",stdin);freopen(#name".out","w",stdout);
#define fre(name) freopen(#name".txt","r",stdin);
#ifdef WIN32
#define LL "%I64d"
#else
#define LL "%lld"
#endif

const ll S=1e6;
const ll N=S+;
const ll mod=1e9+;
ll cas,n,m,ans,dp[N],fz[N],fm[N];
ll fpow(ll a,ll p){
    ll res=;
    for(;p;p>>=,a=a*a%mod) if(p&) res=res*a%mod;
    return res;
}
void first(){
    dp[]=;dp[]=;
    fz[]=fz[]=fm[]=fm[]=;
    for(ll i=;i<=S;i++) dp[i]=(i*dp[i-]%mod+((i&)?-:)+mod)%mod;
    for(ll i=;i<=S;i++) fz[i]=fz[i-]*i%mod;
    fm[S]=fpow(fz[S],mod-);
    for(ll i=S-;i>=;i--) fm[i]=fm[i+]*(i+)%mod;
}
void solve(){
    for(cas=read();cas--;){
        n=read();m=read();
        ans=(((fz[n]*fm[m])%mod*fm[n-m])%mod*dp[n-m])%mod;
        printf(LL "\n",ans);
    }
}
int main(){
    //FRE(permutation);
    first();
    solve();
    return ;
}

征途

/*
ans=m(x_1^2+X_2^2+……+x_m^2)-(x_1+X_2+……+x_m)^2
令dp[i][j]表示前i个数已经选择到了x_j的最小值；这样就变成了一个区间dp
我懒，就不写单调队列优化了
*/
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;

#define pf(x) ((x)*(x))
#define fro __attribute__((optimize("O2")))
#define FRE(name) freopen(#name".in","r",stdin);freopen(#name".out","w",stdout);
#define fre(name) freopen(#name".txt","r",stdin);
#ifdef WIN32
#define LL "%I64d"
#else
#define LL "%lld"
#endif

const int N=;
int n,m,a[N],p[N][N];
ll ans,sum[N],dp[N][N];

fro int main(){
    //FRE(journey);
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%d",&a[i]),sum[i]=sum[i-]+a[i];
    memset(dp,0x3f3f3f3f,sizeof dp);
    dp[][]=;
    for(int j=;j<=m;j++){
        for(int i=j;i<=n;i++){
            for(int k=p[i-][j];k<i;k++){
                if(dp[i][j]>dp[k][j-]+pf(sum[i]-sum[k])){
                    dp[i][j]=dp[k][j-]+pf(sum[i]-sum[k]);
                    p[i][j]=k;
                }
            }
        }
    }
    ans=dp[n][m]*(ll)m-pf(sum[n]);
    printf(LL,ans);
    return ;
}