程序员的算法趣题Q46: 唯一的OX序列

最新推荐文章于 2024-11-06 11:05:01 发布

笨牛慢耕

最新推荐文章于 2024-11-06 11:05:01 发布

阅读量830

点赞数 2

分类专栏：程序员的算法趣题文章标签：算法 python 算法趣题

本文链接：https://blog.csdn.net/chenxy_bwave/article/details/120535400

版权

程序员的算法趣题专栏收录该内容

74 篇文章 13 订阅

订阅专栏

本文介绍了如何解决一个关于N×N矩阵中O和X排列的问题，通过将O和X转化为二进制矩阵并使用哈希表计数signature的方法，计算出满足条件的排列数量。作者提供了Python代码实现，并讨论了可能的优化方案，如位操作和排除特定矩阵。在N=4的情况下，得到了6902种排列。文章还提及后续的优化和相关算法趣题系列。

摘要由CSDN通过智能技术生成

1. 问题描述

当n=4时，像上述例子一样，根据统计结果重新排列O和X的位置，只有一种排列方式的O和X的排列一共有多少种呢？

2. 解题分析

因为是对O计数，可以用1代表O，用0代表x，这样原矩阵就转化为一个二进制矩阵。

以下采用暴力搜索法。

对N*N的所有可能的二进制矩阵进行N行和N列的，所得的2*N个值形成的排列{r1_sum, r2_sum, …,rN_sum, c1_sum, c2_sum, …, cN_sum }构成这个矩阵的signature。然后查询值对应唯一的矩阵的signature的个数。可以在遍历所有矩阵时，对各种signature出现的次数进行计数，最后计数值为1的signature个数即为所求结果。signature出现的次数可以用哈希表来存储，在python中就是dict()。

N*N的所有可能的二进制矩阵种类数为 , N=4时为65536，随着N增大急剧增大。

算法流程如下所示：

3. 代码及测试

# -*- coding: utf-8 -*-
"""
Created on Wed Sep 29 07:51:03 2021

@author: chenxy
"""

import sys
import time
import datetime
import math
# import random
from   typing import List
# from   queue import Queue
from   collections import deque
import itertools as it
import numpy as np

N = 4
sigCount = dict()

tStart = time.perf_counter()
for node in it.product([0,1],repeat=N**2):
    a = np.array(node).reshape(N,N)
    # print(a)
    col_sum = np.sum(a,axis=0)
    row_sum = np.sum(a,axis=1)
    sig = tuple(np.concatenate((col_sum,row_sum)))
    if sig in sigCount:
        sigCount[sig] += 1
    else:
        sigCount[sig]  = 1

count = 0
for key in sigCount:
    if sigCount[key] == 1:
        count += 1
tCost = time.perf_counter() - tStart

print('N = {0}, count={1}, tCost = {2:6.3f}(sec)'.format(N,count,tCost))

运行结果：N = 4, count=6902, tCost = 0.891(sec)