HDU 5745 La Vie en rose(DP+bitset优化)

来源:互联网 发布:mac hdmi声音输出设置 编辑:程序博客网 时间:2024/05/16 12:23

[题目链接]

[题意]
给定模式串和主串,模式串可以交换相邻位子的字符,且可以交换多个位子,但每个字符只能被交换一次。问模式串能否通过该变换与主串匹配?输出主串中每个位子的匹配结果,1为可匹配,0为不可匹配。

[分析]
赛上被这道题坑哭了,看到那么多队过自己却没有思路,好不容易想到个方法把样例过了却T了。最后走投无路写了个暴力就A了,1500ms,吓死。
不过赛后貌似加强了数据,卡了暴力,丧病。
不过趁机学习了下bitset优化的方法,还不错

dp公式很容易得到
设主串为s,模式串为t,dp[i][j][k]表示主串到i,模式串到j,模式串当前位置的状态为k(0:不动,1:与前面交换,2:与后面交换)是否匹配,则:
dp[i][j][0] = (dp[i-1][j-1][0] || dp[i-1][j-1][1] ) && (s[i] == t[j])
dp[i][j][1] = (dp[i-1][j-1][2] ) && ( s[i-1] == t[j] )
dp[i][j][2] = (dp[i-1][j-1][0] || dp[i-1][j-1][1] )&& (s[i+1] == t[j] )

优化1:
第二维j实际上只与j-1有关,可以滚动
优化2:
所有值都是bool类型,可以用bitset压位,通过bitset之间的位运算,常数优化

其实听了这些对不会玩bitset的来说还是一脸懵逼(比如我
所以具体一点吧

bitset<N> dp[2][3] , ch[26] ;

首先需要这些bitset,前者当然是存dp值,2用于滚动,3存三种状态;后者对应26个字母是否在主串中的某个位子出现,比如ch[0][5]表示字母a是否出现在主串中的第5个位子

dp[cur][1] = (dp[cur^1][2] << 1) & ch[t[i-1]-'a'] ;

转移的时候这样搞,这里只列出了k=1的转移写法
由于两个bitset可以直接进行位运算,相当于一次运算直接把1-n的所有dp值全部算出,复杂度却只有O(n/w)(w是机器的字节长)
所以预处理每个字符是否在每个位子出现,保存在bitset中,就是为了这里的操作
还有一个疑问,为何要<<1?
很简单,因为i由i-1推得,所以左移1使得对应的位对齐,再运算
这样写刚好卡过

[代码]

#include <bits/stdc++.h>using namespace std ;const int N = 100000 + 5 ;typedef long long LL ;int T , n , m ;char s[N] , t[N] ;bitset<N> dp[2][3] , ch[26] ;int main(){    scanf( "%d" , &T ) ;    while( T-- )    {        scanf( "%d%d" , &n , &m ) ;        scanf( "%s%s" , s+1 , t+1 ) ;        for( int i = 0 ; i < 26 ; i++ )            ch[i].reset() ;        for( int i = 1 ; i <= n ; i++ )            ch[s[i]-'a'].set(i) ;        int cur = 0 ;        dp[cur][0].set() ;        dp[cur][1].reset() ;        dp[cur][2].reset() ;        for( int i = 1 ; i <= m ; i++ )        {            cur ^= 1 ;            for( int j = 0 ; j < 3 ; j++ )                dp[cur][j].reset() ;            dp[cur][0] = ((dp[cur^1][0] | dp[cur^1][1]) << 1) & ch[t[i]-'a'] ;            if( i > 1 )                dp[cur][1] = (dp[cur^1][2] << 1) & ch[t[i-1]-'a'] ;            if( i < m )                dp[cur][2] = ((dp[cur^1][1] | dp[cur^1][0]) << 1) & ch[t[i+1]-'a'] ;        }        for( int i = m ; i <= n ; i++ )            s[i-m] = '0' + (dp[cur][0][i] | dp[cur][1][i]) ;        for( int i = n-m+1 ; i < n ; i++ )            s[i] = '0' ;        s[n] = 0 ;        puts(s) ;    }    return 0 ;}
0 0