ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Solution -「NOI 2021」「洛谷 P7740」机器人游戏

2021-08-08 20:01:54  阅读:168  来源: 互联网

标签:P7740 const NOI int mathcal 洛谷 MOD inline sta


\(\mathcal{Description}\)

  Link.

  自己去读题面叭~

\(\mathcal{Solution}\)

  首先,参悟【样例解释 #2】。一种暴力的思路即为钦定集合 \(S\) 内的位置都合法,容斥计数。其中对于每条纸带的每个位置,有三种情况(令 _ 为“保持不变”,注意没有被机器人经过的位置都有这种修改):

  • 同时存在 _*;或者同时存在 01:只能为空,方案数为 \(1\);
  • 否则,存在(_*)且存在(01):只能为空或 01 中的一个,方案数为 \(2\);
  • 否则,方案数为 \(3\)。

  暴力做的复杂度是 \(\mathcal O(2^nn^am^b)\),\(a,b\) 看细节实现,不太可观。

  尝试钦定最后一个在 \(S\) 中的位置为 \(r\),那么所有修改长度大于 \(n-r+1\) 的机器人所在的纸带都必须全空。在此限制下带着容斥系数状压 DP,令 \(f(i,S)\) 表示考虑了纸带的位置 \(i\),前若干个位置的选择情况为 \(S\),注意此处 \(|S|\le2^{\min\{i,n-r\}}=\mathcal O(2^{n/2})\),转移时可以 \(\mathcal O(nm)\) 暴力计算当前位置的贡献,得到了 \(\mathcal O(2^{n/2}nm)\) 的算法。

  顺带,注意到统计当前位置贡献时可以描述为形如“有多少个机器人的修改序列中,下标属于 \(S\) 的位置上存在 _/*/0/1”,可以用 std::bitset 优化。最终复杂度为 \(\mathcal O(\frac{2^{n/2}nm}{\omega})\)。

\(\mathcal{Code}\)

/*~Rainybunny~*/

#include <bitset>
#include <cstdio>
#include <cassert>
#include <cstring>

#define rep( i, l, r ) for ( int i = l, rep##i = r; i <= rep##i; ++i )
#define per( i, r, l ) for ( int i = r, per##i = l; i >= per##i; --i )

typedef unsigned long long ULL;

const int MAXN = 32, MAXM = 1e3, MOD = 1e9 + 7;
int n, m, pw2[MAXM + 5], pw3[MAXM + 5];

inline int imin( const int a, const int b ) { return a < b ? a : b; }
inline void subeq( int& a, const int b ) { ( a -= b ) < 0 && ( a += MOD ); }
inline int mul( const int a, const int b ) { return int( 1ll * a * b % MOD ); }
inline int add( int a, const int b ) { return ( a += b ) < MOD ? a : a - MOD; }
inline void addeq( int& a, const int b ) { ( a += b ) >= MOD && ( a -= MOD ); }

struct Robot {
	int len, val[MAXN + 5];
	inline int operator [] ( const int k ) const {
		return k <= len ? val[k] : 0;
	}
	inline void read() {
		static char str[105]; scanf( "%s", str );
		val[len = 1] = 0;
		for ( int i = 0; str[i]; ++i ) {
			if ( str[i] == 'R' ) val[++len] = 0;
			else if ( str[i] == '0' ) val[len] = 1;
			else if ( str[i] == '1' ) val[len] = 2;
			else val[len] = 3 - val[len];
		}
	}
} rbt[MAXM + 5];

inline int solve( const int r ) {
	static int f[2][1 << 17][2];
	static std::bitset<MAXM + 5> sig[MAXN + 5][4], g[1 << 17][4];
	int allc = 0;
	
	rep ( i, 0, n - 1 ) rep ( j, 0, 3 ) sig[i][j].reset();
	rep ( i, 1, m ) if ( rbt[i].len <= n - r + 1 ) {
		++allc;
		rep ( j, 0, n - 1 )
			sig[j][rbt[i][j + 1]].set( i );
	}
	rep ( i, 0, imin( n - r, r ) ) rep ( j, 0, 3 ) {
		g[1 << i][j] = sig[i][j];
	}
	
	rep ( S, 1, ( 1 << imin( n - r, r ) << 1 ) - 1 ) if ( S & ( S - 1 ) ) {
		rep ( i, 0, 3 ) {
			g[S][i] = g[S ^ ( S & -S )][i] | g[S & -S][i];
		}
	}
	
	auto calc = [&]( const bool sty, const ULL S )->int {
		static std::bitset<MAXM + 5> x, y;
		x = ( sty ? g[S][3] : g[S][0] & g[S][3] ) | ( g[S][1] & g[S][2] );
		y = ~x & ( sty ? g[S][1] | g[S][2] :
		  ( g[S][0] | g[S][3] ) & ( g[S][1] | g[S][2] ) );
		int t = y.count();
		return mul( pw2[t], pw3[allc - t - x.count()] );
	};
	
	memset( f[0], 0, sizeof f[0] ), f[0][0][0] = 1;
	for ( int i = 0, sta = 0; i < r; ++i, sta ^= 1 ) {
		memset( f[!sta], 0, sizeof f[!sta] );
	  	rep ( S, 0, ( 1 << imin( i, n - r ) ) - 1 ) {
			ULL T = ULL( S ) << 1ull & ( ( 1ull << n >> r ) - 1 );
			bool flg = ULL( S ) << 1ull << r >> n & 1ull;
			
			rep ( t, 0, 1 ) {
				int cur = f[sta][S][t]; f[sta][S][t] = 0;
				if ( !cur ) continue;
				subeq( f[!sta][r == n ? 0 : T | 1][t || flg || r == n],
				  mul( cur, calc( t || i + 1 < r, S << 1 | 1 ) ) );
				if ( i + 1 < r ) {
					addeq( f[!sta][T][t || flg],
					  mul( cur, calc( 1, S << 1 ) ) );
				}
			}
		}
	}

	int ret = 0;
	rep ( t, 0, 1 ) {
		for ( int d = 1, sta = r & 1; d <= n - r; ++d, sta ^= 1 ) {
			rep ( i, 0, imin( r, n - r ) - 1 ) {
				rep ( j, t, 3 ) g[1 << i][j] = sig[i + d][j];
			}
			rep ( S, 0, ( 1 << imin( r, n - r ) ) - 1 ) {
				if ( S & ( S - 1 ) ) {
					rep ( j, t, 3 ) {
						g[S][j] = g[S ^ ( S & -S )][j] | g[S & -S][j];
					}
				}
				f[!sta][S][t] = mul( f[sta][S][t], calc( t, S ) );
				f[sta][S][t] = 0;
			}
		}
		rep ( S, 0, ( 1 << imin( r, n - r ) ) - 1 ) {
			addeq( ret, f[n & 1][S][t] );
		}
	}
	return ret;
}

int main() {
    // freopen( "robot.in", "r", stdin );
    // freopen( "robot.out", "w", stdout );
    
	scanf( "%d %d", &n, &m );
	pw2[0] = pw3[0] = 1;
	rep ( i, 1, m ) {
		rbt[i].read();
		pw2[i] = mul( 2, pw2[i - 1] ), pw3[i] = mul( 3, pw3[i - 1] );
	}
	
	int ans = 0;
	rep ( i, 1, n ) subeq( ans, solve( i ) );
	printf( "%d\n", ans );
	return 0;
}

标签:P7740,const,NOI,int,mathcal,洛谷,MOD,inline,sta
来源: https://www.cnblogs.com/rainybunny/p/15115936.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有