P2915 [USACO08NOV] Mixed Up Cows

题目描述

Each of Farmer John's N (4 <= N <= 16) cows has a unique serial number S_i (1 <= S_i <= 25,000). The cows are so proud of it that each one now wears her number in a gangsta manner engraved in large letters on a gold plate hung around her ample bovine neck.

Gangsta cows are rebellious and line up to be milked in an order called 'Mixed Up'. A cow order is 'Mixed Up' if the sequence of serial numbers formed by their milking line is such that the serial numbers of every pair of consecutive cows in line differs by more than K (1 <= K <= 3400). For example, if N = 6 and K = 1 then 1, 3, 5, 2, 6, 4 is a 'Mixed Up' lineup but 1, 3, 6, 5, 2, 4 is not (since the consecutive numbers 5 and 6 differ by 1).

How many different ways can N cows be Mixed Up?

For your first 10 submissions, you will be provided with the results of running your program on a part of the actual test data.

POINTS: 200

约翰家有N头奶牛，第i头奶牛的编号是Si，每头奶牛的编号都是唯一的。这些奶牛最近在闹脾气，为表达不满的情绪，她们在挤奶的时候一定要排成混乱的队伍。在一只混乱的队伍中，相邻奶牛的编号之差均超过K。比如当K = 1时，1, 3, 5, 2, 6, 4就是一支混乱的队伍，而1, 3, 6, 5, 2, 4不是，因为6和5只差1。请数一数，有多少种队形是混乱的呢？

输入输出格式

输入格式：

* Line 1: Two space-separated integers: N and K

* Lines 2..N+1: Line i+1 contains a single integer that is the serial number of cow i: S_i

输出格式：

* Line 1: A single integer that is the number of ways that N
cows can be 'Mixed Up'. The answer is guaranteed to fit in a 64 bit
integer.

输入输出样例

输入样例#1：

输出样例#1：

说明

The 2 possible Mixed Up arrangements are:

3 1 4 2

2 4 1 3

干了一下午的线段树合并和主席树还没整明白，想到自己还这么菜就很烦，看到一道状压DP能自己静下心来推了还是挺开心的。

如果$n$小于10就可以dfs回溯求了。

$f\left[ S\right] \left[ i\right]$表示集合$S$已被排列且已第$i$个编号结尾的方案数

$f\left[ S|(1<<i)\right]\left[i \right ] =\sum f\left[ s\right]\left[j \right ]$ 其中$j\in S\& \& \left| s_{i}-s_{j}\right| >k$

复杂度$O\left( n^{2}2^{n}\right)$

#include <bits/stdc++.h>

#define ll long long

using namespace std;

inline int read() {

    int x = , f = ; char ch = getchar();

    while (ch < '' || ch > '') { if (ch == '-') f = -; ch = getchar(); }

    while (ch >= '' && ch <= '') { x = x *  + ch - ; ch = getchar(); }

    return x * f;

}

const int N = ;

ll f[<<N][N];

int t[N], n, k;

int main() {

    n = read(), k = read();

    for (int i = ; i <= n; i++) t[i] = read();

    f[][] = ;

    int S =  << n;

    for (int s = ; s < S; s++) {

        for (int i = ; i <= n; i++) {

            int ii = i - ;

            if (( << ii) & s) continue;

            for (int j = ; j <= n; j++) {

                int jj = j - ;

                if (jj < ) { f[s | ( << ii)][i] += f[s][j]; continue; }

                if (( << jj) & s && abs(t[j] - t[i]) > k) {

                    f[s | ( << ii)][i] += f[s][j];

                }

            }

        }

    }

    ll ans = ;

    for (int i = ; i <= n; i++) ans += f[S - ][i];

    printf("%lld\n", ans);

    return ;

}