Codeforces1472 F. New Year‘s Puzzle(离散化+dp)

题意:

给定n和m,
表示有一个2*n的矩阵,
给定m个坐标,表示这m个坐标被禁止。

现在你要用1x2和2x1的骨牌放满矩阵,问是否可能。

数据范围:n<=1e9,m<=2e5

解法:

容易想到令d[i][j]表示前i-1个格子填满,第i个格子状态为j是否存在.
然后dp一下就行了,
但是这题的n很大,开不下dp数组,
观察到m很小,因此离散化一下坐标,将n缩小为1e5级别的就行了.

code:

#include <bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
struct Node{
    
    
    int r,c;
}e[maxm];
bool d[maxm<<2][4];
bool del[3][maxm<<2];
int cc[maxm];
int n,m;
bool cmp(Node a,Node b){
    
    
    return a.c<b.c;
}
void solve(){
    
    
    cin>>n>>m;
    for(int i=1;i<=m;i++){
    
    
        cin>>e[i].r>>e[i].c;
    }
    //离散化
    sort(e+1,e+1+m,cmp);
    for(int i=1;i<=m;i++){
    
    
        if(e[i].c==e[i-1].c){
    
    
            cc[i]=cc[i-1];
        }else if((e[i].c-e[i-1].c)%2==1){
    
    //与上一个距离为奇数
            cc[i]=cc[i-1]+1;
        }else{
    
    //与上一个距离为偶数
            cc[i]=cc[i-1]+2;
        }
    }
    for(int i=1;i<=m;i++){
    
    
        e[i].c=cc[i];
    }
    //标记删除的格子
    for(int i=0;i<=cc[m];i++){
    
    
        del[1][i]=del[2][i]=0;
    }
    for(int i=1;i<=m;i++){
    
    
        del[e[i].r][e[i].c]=1;
    }
    //初始化dp数组
    for(int i=0;i<=cc[m];i++){
    
    
        for(int j=0;j<4;j++){
    
    
            d[i][j]=0;
        }
    }
    d[0][3]=1;
    //1=10,2=01,3=11
    for(int i=1;i<=cc[m];i++){
    
    
        if(!del[1][i]&&!del[2][i]){
    
    //00
            d[i][1]|=d[i-1][2];
            d[i][2]|=d[i-1][1];
            d[i][3]|=d[i-1][3];
        }else if(del[1][i]&&!del[2][i]){
    
    //10
            d[i][1]|=d[i-1][3];
            d[i][3]|=d[i-1][1];
        }else if(!del[1][i]&&del[2][i]){
    
    //01
            d[i][2]|=d[i-1][3];
            d[i][3]|=d[i-1][2];
        }else if(del[1][i]&&del[2][i]){
    
    //11
            d[i][3]|=d[i-1][3];
        }
    }
    //
    if(d[cc[m]][3]){
    
    
        cout<<"YES"<<endl;
    }else{
    
    
        cout<<"NO"<<endl;
    }
}
signed main(){
    
    
    ios::sync_with_stdio(0);
    int T;cin>>T;
    while(T--){
    
    
        solve();
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_44178736/article/details/112705493