【状压DP】【cofun1895】队伍统计

【cofun1895】队伍统计

Description
现在有n个人要排成一列，编号为1->n 。但由于一些不明原因的关系，人与人之间可能存在一些矛盾关系，具体有m条矛盾关系(u,v),表示编号为u的人想要排在编号为v的人前面。要使得队伍和谐，最多不能违背k条矛盾关系（即不能有超过k条矛盾关系(u,v),满足最后v排在了u前面）。问有多少合法的排列。答案对10^9+7取模。

Input Format
输入文件名为count.in。
第一行包括三个整数n,m,k。
接下来m行，每行两个整数u,v,描述一个矛盾关系(u,v)。
保证不存在两对矛盾关系(u,v),(x,y)，使得u=x且v=y 。
Output Format
输出文件名为count.out。
输出包括一行表示合法的排列数。

Sample Input
输入1：
4 2 1
1 3
4 2

输入2：
10 12 3
2 6
6 10
1 7
4 1
6 1
2 4
7 6
1 4
10 4
10 9
5 9
8 10
Sample Output
输出1：
18
输出2：
123120

Hint
对于30%的数据，n<=10
对于60%的数据，n<=15
对应100%的数据，n,k<=20,m<=n*(n-1)，保证矛盾关系不重复。

---

• 分析：
  n <= 20，每个人最多与19个人关系矛盾，考虑状压DP。

  1. 读取矛盾关系并压进二进制数中，用数组存储。g[i]：第i个人的矛盾关系情况（0：不矛盾/ 1：矛盾）。
  2. 状压DP：
     f[i][l]：队伍情况为i（i为二进制数，0：该标号人不在队伍中/ 1：在），队伍中有l条矛盾关系的排列数。
     q：当前队伍情况与新增的j的矛盾关系条数，用__builtin_popcount()函数算出。
     转移方程：

f[i | (1 << j)][l + q] = (f[i | (1 << j)][l + q] + f[i][l]) % INF;

【考试时没做几题状压，没想到先用二进制压每个人的矛盾关系，预处理超时了3个点【STO。

---

• 代码：

#include <bits/stdc++.h>
 using namespace std;

 const long long INF = 1e9 + 7;
 int n, m, k, i, j, l, q, u, v, head[25], g[25];
 long long f[1 << 20][21], ans;

 inline int read()
 {
    int x = 0, w = 1;
    char ch = 0;
    while(ch < '0' || ch > '9')
    {
        if (ch == '-')
            w = -1;
        ch = getchar();
     }
    while(ch >= '0' && ch <= '9')
          x = x * 10 + ch - '0', ch = getchar();
    return x * w;
  } //读入优化 

 inline void write(long long x)
 {
    if (x < 0)
        putchar('-'), x = -x;
    if (x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
 }//输出优化 

 int main()
 {

    n = read(), m = read(), k = read();
    for(i = 1, j = 0; i <= m; i ++)
    {
        u = read(), v = read();
        g[v - 1] |= 1 << (u - 1);
     }
    //读入并压位1.    
    f[0][0] = 1;
    for(i = 0; i < 1 << n; i ++)
        for(j = 0; j < n; j ++)
        if (! (i & (1 << j)))
        {
            q = __builtin_popcount(i & g[j]);//计算二进制中1的个数 
            for(l = 0; l <= k - q; l ++)
                f[i | (1 << j)][l + q] = (f[i | (1 << j)][l + q] + f[i][l]) % INF;
         }
    //状压DP 2.    
    ans = 0;
    for(i = 0; i <= k; i ++)
        ans = (ans + f[(1 << n) - 1][i]) % INF;
    write(ans);
    //统计答案并输出    
    return 0;

 } 

---

给同学讲寿司晚宴，用QQ聊了好久 _ (:з」∠) _
期间手抖关了一次编辑器【当然没保存，于是码到现在
要加快速度了！加油！！