数据结构练习154--157

7-154 n皇后

n皇后问题是指将 n 个皇后放在 n×n 的国际象棋棋盘上，使得皇后不能相互攻击到，即任意两个皇后都不能处于同一行、同一列或同一斜线上。

输入格式:

共一行，包含整数 n (1 ≤ n ≤ 12)。

输出格式:

给出所有可能摆放情况的种数，结尾无空格换行。

输入样例:

在这里给出一组输入。例如：

4

输出样例:

在这里给出相应的输出。例如：

2

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <unordered_map>

#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)

using namespace std;

const int N = 100;
int a[N]; // i的位置存储j表示：第i行第j个位置有皇后
int n; 
int b1[N], b2[N]; // 可以用数组存储哈希

void print()
{
    for (int i = 1; i <= n; i ++ )
    {
        for (int j = 1; j <= n; j ++ )
        {
            if (a[i] == j) cout << 'Q';
            else cout << '.';
        }
        cout << endl;
    }
}

int main()
{
    IOS;
    cin >> n;

    // 全排列必须升序, 先初始化数组
    for (int i = 1; i <= n; i ++ ) a[i] = i;

    // 枚举全排列, 这样就可以保证皇后行和列不同
    do 
    {
        vector<int> b1(N), b2(N); // 存储截距
        //memset(b1, 0, sizeof b1), memset(b2, 0, sizeof b2);
        
        bool vaild = true;
        for(int i = 1; i <= n; i ++ ) 
        {
            if (++ b1[i - a[i] + n] > 1 | ++ b2[i + a[i]] > 1) // 如果斜率不是第一次出现
            {
                vaild = false;
                break;
            }
        }
        if (vaild) 
        {
            print();
            cout << endl;
        }
    } while (next_permutation(a + 1, a + n + 1));
    
    return 0;
}

7-156 最小支撑树

 给定一个包含n个顶点的正权无向图，编号为1至n。请编写程序求出其最小支撑树，并计算其边权之和。

输入格式:

输入包含多组数据。每组数据第一行为2个整数n和e，均不超过1500，分别表示图的顶点数和边数。接下来e行表示每条边的信息，每行为3个非负整数a、b、c，其中a和b表示该边的端点编号，c表示权值，不超过100。各边并非按端点编号顺序排列。

输出格式:

对于每组数据，若存在最小支撑树则输出一个整数，为最小支撑树各边权值之和；若不存在最小支撑树，则输出“There is no minimum spanning tree.”。

输入样例:

4 5
1 2 1
1 4 1
2 4 5
2 3 1
3 4 8
4 2
1 2 1
3 4 8

输出样例:

3
There is no minimum spanning tree.

 #include <bits/stdc++.h>
using namespace std;

typedef struct{
	int u;//顶点1
	int v;//顶点2 
	int w;//权值 
}Edge;//结构体存边的信息 
int a[201];

bool cmp(Edge edge1,Edge edge2){
	return edge1.w<edge2.w;
}

void init(){//初始化 
	for(int i=0; i<201; i++){
		a[i] = i;
	}
}

int find(int x){//查找 
	int r = x;
	while(a[r]!=r)
		r = a[r];
	return r;
}

int join(int x,int y){//合并 
	int fx = find(x);
	int fy = find(y);
	if(fx!=fy){
		a[fy] = fx;
		return 1;
	}
	return 0;
}

int main(){
    int n,e;
    while(cin >> n >> e){
    	Edge edge[e];
    	init();
    	for(int i=0; i<e; i++){
    		cin >> edge[i].u >> edge[i].v >> edge[i].w;
		}
		sort(edge,edge+e,cmp); //边按权值从小到大排序 
		int cnt=0, w_sum=0; //边数、权值和
		for(int i=0; i<e; i++){
			if(join(edge[i].u,edge[i].v)){//如果顶点u和v不在一个连通图里
				cnt++;
				w_sum += edge[i].w;
			}
			if(cnt==n-1) //边数 = 顶点数-1，找到最小生成树 
				break;
		}
		if(cnt==n-1)
			cout << w_sum << endl;
		else cout << "There is no minimum spanning tree." << endl; 
	}
    
    return 0;
}

 

 7-157 选数

 已知n个整数x1,x2,x3...xi，以及1个整数k(k<n)。从 n 个整数中任选 k个整数相加，可分别得到一系列的和。例如当 n=4，k=3，4个整数分别为3,7,12,19 时，可得全部的组合与它们的和为：
3+7+12=22，
3+7+19=29，
7+12+19=38，
3+12+19=34，

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=29

输入格式:

第一行两个空格隔开的整数 n,k（1≤n≤20，k<n）
第二行n个整数，两数之间空格隔开（1≤xi≤1000000）

输出格式:

输出一个整数，表示种类数。

输入样例:

在这里给出一组输入。例如：

4 3
3 7 12 19

输出样例:

在这里给出相应的输出。例如：

1

#include<stdio.h>
int prime(int n);                       //判断是否为素数
void fun(int x,int sum,int l);          //x表示累加的个数,sum表示累加和,l表示选择数字的位置
int n,k,count = 0,num[20];
int main()
{

	int i;
	scanf("%d%d",&n,&k);
	for(i = 0 ; i < n ; i++)
		scanf("%d",&num[i]);
	fun(0,0,0);
	printf("%d\n",count);
	return 0;
}

int prime(int n)
{
	int i;
	if(n <=1 ) return 0;
	for(i = 2 ; i*i <= n ; i++)
	{
		if(n%i == 0) return 0;
	}
	return 1;
}

void fun(int x,int sum,int l)
{
	int i;
	if(x == k)                      //达到k个,判断是否为素数
	{
		if(prime(sum))
			count++;
	}
	for(i = l ; i < n ; i++)        //没到k个,继续加下一个数
		fun(x+1,sum+num[i],i+1);    //核心:相当于填空
}