F-Military Class

https://codeforces.com/gym/324833/problem/F

发现该题相邻的范围e<=4，所以可以状压dp
方法是遍历第一排的每个士兵，分别与第二列士兵进行匹配

dp[i][j]中
i代表第一队的第几个士兵
j代表第二队在e范围内（也就是相邻）的士兵是否已经匹配
则可以通过j不断向右滚动来传递之前已经匹配好的情况

对于每一次dp：
先查看选取的位置是否在边界内
然后查看是否可以匹配该位置
最后查看该位置是否已经被占用了

#include<bits/stdc++.h>
using namespace std;
long long mod=1e9+7;
long long ans=0;
long long dp[2021][(1<<10)];
int p2[20];     //存2的i次方
bool ok[2005][2005];    //记录不能匹配的士兵
int n,e,k,t1,t2;
map<int,int> m;
void init(){    //预处理2的n次方
    for(int i=0;i<=10;i++){
        p2[i+1]=1<<i;   //p1存...000001 p2存...000010 p3存...000100
    }   //p2[n]代表从左向右第n个位置进行匹配
        //因为dp过程中必须右移，所以这里二进制01序列只能存反向的，有点难受
    return;
}
bool check(int i,int j,int k){  
    //i为查找到的士兵位置，j为该士兵另一队伍对应相邻的匹配情况，k为在第几个位置匹配
    if(i+k-e<2)return false;     //左边界限定
    if(i+k-e>n+1)return false;   //右边界限定
    if(j&p2[k])return false;    //查看该点是否已经匹配
    if(ok[i][i-e-1+k])return false; //查看是否可以匹配
    return true;
}
int main(){
    init();
    cin>>n>>e>>k;
    for(int i=1;i<=k;i++){
        cin>>t1>>t2;
        ok[t1][t2]=1;
    }
    dp[1][0]=1;
    for(int i=1;i<=n;i++){  //枚举第一队每个士兵
        for(int j=0;j<p2[2*(e+1)];j++){     //枚举第一队某士兵对应第二队的相邻的情况
            for(int k=1;k<=(2*e+1);k++){    //枚举匹配位置（使用的01序列是反向的）
                if(check(i,j,k)){
                    dp[i+1][(j|p2[k])>>1]+=dp[i][j];
                    dp[i+1][(j|p2[k])>>1]%=mod;
                    //这里犯了一个智障错误导致多调了一个多小时！！！
                    //※注意！！！！这里必须右移！！！！
                    //如果向左移，最左边的1会超出去，导致范围跑到了外面，后面就没法传递了
                }
            }
        }
    }
    for(int i=0;i<p2[2*(e+1)];i++){
        ans+=dp[n+1][i];
        //因为添加了check限制
        //所以其实只有一个dp[n+1]不是0
        //这个dp[n+1]内容就是答案
        //但是找出来是哪个挺麻烦的
        //不如干脆全加起来
    }
    cout<<ans;
    return 0;
}