「JOI Open 2016」摩天大楼(笛卡尔树dp+优化)

https://loj.ac/problem/2743

先排序,从小到大考虑。

最暴力直接设\(f[i][j][k][u]\)表示前i个数,分成j段有顺序的,和差一共为k,首尾确定了u个的方案数。

复杂度是\(O(n^3*A)\)

没有用到\(L<=1000\)

假设在每一段的附近都加上了\(a[i]\),可以得到这个和是非递减的,每次加\(+(∆a)*(2j-u)\)

最后这个和就是实际的相邻差,所以它要一直\(<=L\),复杂度\(O(n^2L)\)

#include<bits/stdc++.h>
#define fo(i, x, y) for(int i = x, _b = y; i <= _b; i ++)
#define ff(i, x, y) for(int i = x, _b = y; i <  _b; i ++)
#define fd(i, x, y) for(int i = x, _b = y; i >= _b; i --)
#define ll long long
#define pp printf
#define hh pp("\n")
#define min(a, b) ((a) < (b) ? (a) : (b))
#define max(a, b) ((a) > (b) ? (a) : (b))
using namespace std;

const int mo = 1e9 + 7;

const int N = 105, M = 1005;

int n, m, a[N];

ll f[2][N][M][3], o;

void add(int j, int k, int u, ll q) {
    f[!o][j][k][u] += q;
}

int main() {
    scanf("%d %d", &n, &m);
    fo(i, 1, n) scanf("%d", &a[i]);
    if(n == 1) {
        pp("1\n"); return 0;
    }
    sort(a + 1, a + n + 1);
    f[o][0][0][0] = 1;
    fo(i, 1, n) {
        memset(f[!o], 0, sizeof f[!o]);
        int det = a[i] - a[i - 1], p = a[i];
        fo(j, 0, i) fo(k, 0, m) fo(u, 0, 2) if(f[o][j][k][u]) {
            f[o][j][k][u] %= mo;
            int nk = k + det * (2 * j - u);
            if(nk > m) continue;
            ll q = f[o][j][k][u];
            add(j + 1, nk, u, q * (j - u + 1));
            if(u < 2) add(j + 1, nk, u + 1, q * (1 + (!u)));
            if((j > u || i == n) && u < 2) add(j, nk, u + 1, q * (1 + (!u)));
            add(j, nk, u, q * (2 * j - u));
            if(j > 1 && !(j == 2 && u == 2 && i < n))
                add(j - 1, nk, u, q * (j - 1));
        }
        o = !o;
    }
    ll ans = 0;
    fo(k, 0, m) ans = (ans + f[o][1][k][2]) % mo;
    pp("%lld\n", ans);
}

猜你喜欢

转载自www.cnblogs.com/coldchair/p/12207336.html