Jumping on Walls CodeForces

Jumping on Walls CodeForces - 198B

应该是一个隐式图的bfs，或者叫dp。

先是一个TLE的O(nklogn)

#include<cstdio>
#include<set>
using namespace std;
typedef pair<bool,int> P;
set<P> ss[2];
P t;
char s[2][100100];
int n,k,ii;
int main()
{
    int i,num,hei;
    scanf("%d%d",&n,&k);
    scanf("%s",s[0]+1);
    scanf("%s",s[1]+1);
    if(s[0][1]!='X')    ss[0].insert(P(0,1));
    for(i=1;i<=n;i++)
    {
        ii^=1;
        ss[ii].clear();
        for(auto t:ss[ii^1])
        {
            num=t.first;
            hei=t.second;
            if(hei+k>n)
            {
                puts("YES");
                return 0;
            }
            if(hei-1>i&&s[num][hei-1]!='X')
                ss[ii].insert(P(num,hei-1));
            if(hei+1>i&&s[num][hei+1]!='X')
                ss[ii].insert(P(num,hei+1));
            if(hei+k>i&&s[num^1][hei+k]!='X')
                ss[ii].insert(P(num^1,hei+k));
        }
    }
    puts("NO");
    return 0;
}

后来意识到了同样的位置，在较早的时间到过之后在较晚的时间再到那里一定不会比较早的时间更好，因此相同状态只需遍历一次，可以把复杂度优化到O(nlogn)（话说这不是显而易见吗，怎么就没有想到呢）

#include<cstdio>
#include<set>
using namespace std;
typedef pair<bool,int> P;
set<P> ss[2];
P t;
char s[2][100100];
int n,k,ii;
bool vis[2][100100];
int main()
{
    int i,num,hei;
    scanf("%d%d",&n,&k);
    scanf("%s",s[0]+1);
    scanf("%s",s[1]+1);
    if(s[0][1]!='X')    ss[0].insert(P(0,1));
    for(i=1;i<=n;i++)
    {
        ii^=1;
        ss[ii].clear();
        for(auto t:ss[ii^1])
        {
            num=t.first;
            hei=t.second;
            if(hei+k>n)
            {
                puts("YES");
                return 0;
            }
            if(hei-1>i&&s[num][hei-1]!='X'&&(!vis[num][hei-1]))
                ss[ii].insert(P(num,hei-1)),vis[num][hei-1]=1;
            if(hei+1>i&&s[num][hei+1]!='X'&&(!vis[num][hei+1]))
                ss[ii].insert(P(num,hei+1)),vis[num][hei+1]=1;
            if(hei+k>i&&s[num^1][hei+k]!='X'&&(!vis[num^1][hei+k]))
                ss[ii].insert(P(num^1,hei+k)),vis[num^1][hei+k]=1;
        }
    }
    puts("NO");
    return 0;
}

upd20190310:

啧，貌似我以前假了。直接ans[i][j]表示到达i墙j位置的最小时间，如果在这个时间水位已经没过这个点了那么这个点就是废的，不要从这个点去更新其他点

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
typedef long long ll;
typedef unsigned long long ull;
struct P
{
    int x,y;
};
queue<P> q;
int n,K;
int ans[2][100011];
char s[2][100011];
char v1[2][100011];
int main()
{
    P t;int x,y;
    scanf("%d%d",&n,&K);
    scanf("%s",s[0]+1);
    scanf("%s",s[1]+1);
    memset(ans,0x3f,sizeof(ans));
    ans[0][1]=0;v1[0][1]=1;q.push((P){0,1});
    while(!q.empty())
    {
        t=q.front();q.pop();
        x=t.x;y=t.y;
        if(y<=ans[x][y])    continue;
        if(s[x][y]=='X')    continue;
        if(y+1>n || y+K>n)
        {
            puts("YES");
            return 0;
        }
        if(y>1 && !v1[x][y-1])
        {
            v1[x][y-1]=1;ans[x][y-1]=ans[x][y]+1;
            q.push((P){x,y-1});
        }
        if(!v1[x][y+1])
        {
            v1[x][y+1]=1;ans[x][y+1]=ans[x][y]+1;
            q.push((P){x,y+1});
        }
        if(!v1[x^1][y+K])
        {
            v1[x^1][y+K]=1;ans[x^1][y+K]=ans[x][y]+1;
            q.push((P){x^1,y+K});
        }
    }
    puts("NO");
    return 0;
}