[AT2669] [AGC017 F] Zigzag

洛谷传送门

Atcoder传送门

题目大意

你有一个由$\frac{n(n+1)}{2}$个点组成的等边三角形。 在$(i,j)$点左下的是$(i+1,j)$点， 右下的是$(i+1,j+1)$点。

现在你要在上面画$m$条从$(1,1)$开始， $(n,p)$结束的连续路径$(p\in [1,n])$， 满足对于任意$1\le i<j\le m$， 第$j$条线的任意一个部分不在第$i$条线的左边。

另外， 你还有$k$个限制需要满足， 第$i$个限制形如$A_i,B_i,C_i$， 表示第$A_i$条路径， 在第$B_i$次决策的时候， 如果$C_i=0$则必须走左下， 否则必须走右下。

现在你想要求出一共有多少种不同的画路径的方案满足以上的要求。

输入输出格式

输入格式

第一行三个正整数$n,m,k$。

以下$k$行， 每行三个正整数$A_i,B_i,C_i$。

输出格式

输出总的方案数$mod1000000007$

输入输出样例

输入样例#1：

3 2 1
1 2 0

输出样例#1：

6

输入样例#2：

3 2 2
1 1 1
2 1 0

输出样例#2：

0

输入样例#3：

5 4 2
1 3 1
4 2 0

输出样例#3：

172

输入样例#4：

20 20 0

输出样例#4：

881396682

解题分析

有个很暴力的状压$DP$， 直接大力枚举状态转移， 复杂度是$m{4}^n$的。

肯定我们枚举两个状态是不科学的， 因为实在有很多不满足条件的状态全被枚举到了。

如果我们设$0$为向左走， $1$为向右走， 不考虑方向的限制，合法转移后的状态一定是满足任何一个前缀$1$的个数都要$>$之前状态前缀$1$的个数。

那么就有另一种$DP$了。 设$dp[i][j][k]$表示考虑到第$i$条路径的第$j$次转移， 转移的状态的前$j$位和$k$是相同的， 那么考虑向右的情况， 直接把后面的第一个$1$变成$0$， 然后把它拿到前面来， 这样中间的一段取$1$取$0$一定都是满足要求的， 不会出现向左越过原来那一条的情况。然后第一维是可以滚动的， 省掉。

总复杂度$O(nm{2}^n)$。

代码如下：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <cstring>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define ll long long
#define MOD 1000000007
#define lbt(i) ((i) & (-(i)))
#define MX 21
int n, m, k, all, last = 1, cur = 0;
int dir[MX][MX], dp[2][1 << MX];
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
IN void ad(int &x, R int v) {x += v; if (x >= MOD) x -= MOD;}
int main(void)
{
	int x, y, foo, bar, ans = 0;
	in(n), in(m), in(k), --n; all = (1 << n) - 1;
	for (R int i = 1; i <= k; ++i)
	in(x), in(y), in(foo), dir[x][y - 1] = foo + 1;
	dp[0][0] = 1;
	for (R int i = 1; i <= m; ++i)
	for (R int j = 0; j < n; ++j)
	{
		std::swap(last, cur); foo = 1 << j, bar = all ^ ((foo << 1) - 1);
		for (R int s = 0; s <= all; ++s) dp[cur][s] = 0;
		for (R int s = 0; s <= all; ++s)
		{
			if (dp[last][s])
			{
				if (dir[i][j] != 2 && (!(foo & s)))//go left
				ad(dp[cur][s], dp[last][s]);
				if (dir[i][j] != 1)//go right
				{
					if (foo & s) x = s;//must go right
					else
					{
						if (!(bar & s)) x = s | foo;//no more bit 1
						else x = (s | foo) ^ lbt(bar & s);//bit 1 exists in higher bits
					}
					ad(dp[cur][x], dp[last][s]);
				}
			}
		}
	}
	for (R int i = 0; i <= all; ++i) ad(ans, dp[cur][i]);
	printf("%d", ans);
}