CodeForces - 1391D - 505 dp

CodeForces - 1391D - 505 dp

题意：给一个矩形，其偶数的子正方形中1的个数必须为奇数

思路：由于长为2的正方形中1的个数为奇数，所以4个长为2拼起来的长度为4的正方形中1的个数一定为偶数，所以当n>=4时，情况必然不存在；当n=1时，输出0；问题转化为求n=2和n=3时的两种情况

做法：

• 当n=2时，枚举第一列的状态，如果第一列中有1个1，那么第二列要么全0要么全1；如果第一列中全1或全0，则第二列一定是1个1。就两种状态，暴力比大小
• 当n=3时，第i+1列的一种状态对应第i列的两种状态，比如000对应101和010，用状压把他们记录成数组，然后线性dp递推过去即可

代码：

const int maxn=1e6+7;
const int INF=1e9;
const ll INFF=1e18;
int n,m,ans=INF,arr[4][maxn],dp[maxn][9];
int g[8][3]=
{
    {5,2},{4,3},{7,0},{1,6},{1,6},{0,7},{4,3},{5,2}
};
int check(int x,int pos)
{
    int x1=(x/4)^arr[1][pos];
    int x2=((x%4)/2)^arr[2][pos];
    int x3=(x%2)^arr[3][pos];
    return x1+x2+x3;
}
int main()
{
    scanf("%d%d",&n,&m);
    if (n>=4)W(-1);
    else if (n==1)W(0);
    else
    {
        rep(i,1,n)rep(j,1,m)scanf("%1d",&arr[i][j]);
        if (n==2)
        {
            int cnt1=0,cnt2=0;
            for (int j=1;j<=m;j++)
            {
                int x=arr[1][j]+arr[2][j];
                if (j%2==1)x!=1?cnt1++:cnt2++;
                else x==1?cnt1++:cnt2++;
            }
            W(min(cnt1,cnt2));
        }
        else
        {
            repp(j,0,8)dp[1][j]=check(j,1);
            rep(i,2,m)
                repp(j,0,8)
                    dp[i][j]=check(j,i)+min(dp[i-1][g[j][0]],dp[i-1][g[j][1]]);
            repp(j,0,8)ans=min(ans,dp[m][j]);
            W(ans);
        }
    }
    return 0;
}