524 - Prime Ring Problem

Problem.png

直接生成所有排列再判断的话会超时,因此需要递归回溯剪枝。
每确定排列中的一位数,就要检查相邻两位的和是否为素数,如果是素数,才继续递归,否则返回上层调用。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>

using namespace std;

bool isprime(int num) {
    for (int i = 2; i <= round(sqrt(num)); i++) {
        if (num % i == 0) {
            return false;
        }
    }
    return true;
}

void print_permutation(int n, int* A, int cur) {
    if (cur == n) {
        // 所有相邻数的和都为素数时
        // 才将排列打印出来
        if (isprime(A[cur - 1] + A[cur % n])) {
            cout << A[0];
            for (int i = 1; i < n; i++) {
                cout << ' ' << A[i];
            }
            cout << endl;
        }
    }
    else {
        for (int i = 2; i <= n; i++) {
            bool ok = true;
            for (int j = 1; j < cur; j++) {
                if (i == A[j]) {
                    ok = false;
                    break;
                }
            }
            if (ok) {
                A[cur] = i;
                // 递归回溯剪枝
                // 每确定排列中的一位数,就要检查相邻两位的和是否为素数
                // 如果是素数,才继续递归,否则返回上层调用
                if (isprime(A[cur - 1] + A[cur])) {
                    print_permutation(n, A, cur + 1);
                }
            }
        }
    }
}

int main() {
    int n;
    int t = 0;
    while (cin >> n) {
        int A[20] = {0};
        A[0] = 1;
        if (t) {
            printf("\n");
        }
        t++;
        printf("Case %d:\n", t);
        print_permutation(n, A, 1);
    }
    return 0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 17,962评论 2 36
  • 国家电网公司企业标准(Q/GDW)- 面向对象的用电信息数据交换协议 - 报批稿:20170802 前言: 排版 ...
    庭说阅读 11,219评论 6 13
  • 早上的公交车上看到一位时髦大姐,黄色小卷发,脸上画了漂亮的妆,耳朵上飘荡的长长的耳环,黑白色的条纹套裙,大红色的尖...
    朝7不晚5阅读 192评论 0 1
  • 同学度假去了,请我替她讲一堂课,我心里觉得大学生的大课我也讲得多了,这总能应付的吧?直播从来没有试过,很有些新鲜感...
    诸慧的身心园地阅读 278评论 4 6
  • 写在前面的话 健身的人很多,网络上也充斥着健身资讯,但是健不好身的人却大有人在,你真的知道健身是什么吗? 小黑随机...
    陈雅诗Nicole阅读 405评论 3 8