码迷,mamicode.com
首页 > 其他好文 > 详细

HihoCoder 1596 Beautiful Sequence题解

时间:2017-10-06 10:37:01      阅读:123      评论:0      收藏:0      [点我收藏+]

标签:int   排序   names   namespace   main   amp   ext   long   [1]   

题意:

首先定义一个序列为Beautiful为:对 2<=i<=n-1  : a[i-1]+a[i+1] >= 2*a[i]

给定n个数,问这些数的所有排列为Beautiful的有多少个

 

题解:

由 a[i-1]+a[i+1] >= 2*a[i],知 a[i+1]-a[i] >= a[i]-a[i-1],

也就是说,Beautiful的序列先单减,后单增。且相邻两项之间的差在变大。

将这n个数排序。

我们用dp[lastA][preA][lastB][preB]记录下:

当单减部分最大数为lastA,次大数为preA。单增部分最大数为lastB,次大数为preB时,的方案数。

新加入一个元素时,考虑一下放在左边,放在右边。就能转移了。

 

code:

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
#define MOD (1000000007)
int n;
LL dp[62][62][62][62];  //dp[last_A][pre_A][last_B][pre_B]
LL a[62];

int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++) {
        scanf("%lld", &a[i]);
    }
    //------------------------------- init 
    sort(a + 1, a + 1 + n);
    int low = 1; LL tmp = 1;
    for (int i = 2; i <= n; i ++) {
        if (a[i] == a[1]) {
            low ++; tmp = tmp * low % MOD;
        } 
    }
    n = n - (low - 1);
    for (int i = 1; i <= n; i ++) {
        a[i] = a[i + low - 1];
    }
    //-------------------------------solve
    LL ans = 0;
    dp[1][0][1][0] = tmp;
    for (int i = 1; i <= n; i ++) {
        for (int j = 0; j < n; j ++) {
            for (int k = 1; k <= n; k ++) {
                for (int t = 0; t < n; t ++) {
                    int nxt = max(i, k) + 1;
                    if (nxt == n+1) {
                        (ans += dp[i][j][k][t]) %= MOD; continue;
                    }

                    // the next element joins A
                    if ( a[nxt] - a[i] >= a[i] - a[j] || j == 0) 
                        ( dp[nxt][i][k][t] += dp[i][j][k][t] ) %= MOD;
                    // the next element joins B
                    if ( a[nxt] - a[k] >= a[k] - a[t] || t == 0)
                        ( dp[i][j][nxt][k] += dp[i][j][k][t] ) %= MOD;
                }
            }
        }
    }
    //------------------------------------
    cout << ans << endl;
}

HihoCoder 1596 Beautiful Sequence题解

标签:int   排序   names   namespace   main   amp   ext   long   [1]   

原文地址:http://www.cnblogs.com/RUSH-D-CAT/p/7630532.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!