洛谷-P1036-選數(順便講講深搜)

编程DSF發表於2024-08-30

首先我們先講一下深度優先搜尋
深度優先搜尋其實是一種遞迴,主要思想是
「向前走,碰壁就回頭,換一條路走」
那麼它不僅僅可以解決迷宮問題(比如輸入一個地圖,向著可以搜的地方繼續深搜。得出路徑)還可以解決列舉問題,今天我們主講這個

列舉元組

第一個,列舉元組
這種的特點在(你們肯定問我為什麼將講,畢竟這種不分清楚很容易混淆,導致驢唇不對馬嘴)

1.元素可以重複(比如說1,1,1算可用答案)
2.位置交換是一個答案(比如說1,2和2,1算兩個可用答案)

第二個,列舉排列
特點:
1、元素不可以重複

題目描述

$n$ 元組是指由 $n$ 個元素組成的序列。例如 $(1,1,2)$ 是一個三元組、$(233,254,277,123)$ 是一個四元組。

給定 $n$ 和 $k$,請按字典序輸出全體 $n$ 元組,其中元組內的元素是在 $[1, k]$ 之間的整數。

「字典序」是指:優先按照第一個元素從小到大的順序,若第一個元素相同,則按第二個元素從小到大……依此類推。詳情參考樣例資料。

輸入格式

僅一行,兩個正整數 $n, k$。

輸出格式

若干行,每行表示一個元組。元組內的元素用空格隔開。

樣例 #1

樣例輸入 #1

2 3

樣例輸出 #1

1 1
1 2
1 3
2 1
2 2
2 3
3 1
3 2
3 3

樣例 #2

樣例輸入 #2

3 3

樣例輸出 #2

1 1 1
1 1 2
1 1 3
1 2 1
1 2 2
1 2 3
1 3 1
1 3 2
1 3 3
2 1 1
2 1 2
2 1 3
2 2 1
2 2 2
2 2 3
2 3 1
2 3 2
2 3 3
3 1 1
3 1 2
3 1 3
3 2 1
3 2 2
3 2 3
3 3 1
3 3 2
3 3 3

提示

對於 $100%$ 的資料,有 $n\leq 5, k\leq 4$。

答案

解析看註釋)

#include <bits/stdc++.h>

using namespace std;
int n,k;
int a[100];
void dfs(int pos){
    if(pos>n){
        for(int i=1;i<=n;i++){
            cout<<a[i]<<' ';//如果列舉到了足夠的元素,輸出,並結束列舉
        }
        cout<<endl;
        return;
    }
    for(int i=1;i<=k;i++){
        a[pos]=i;//先確定第pos個元素,然後繼續列舉
        dfs(pos+1);
    }
}
int main()
{
   
   cin>>n>>k;
   dfs(1);
   return 0;
}
//https://www.luogu.com.cn/problem/B3621

列舉排列

特點:
1.不可以重複 (1,1)不算
2.變種,你可以讓他交換也不算(1,2)(2,1)算一個
具體可以看看列舉排列
程式碼很簡單隻需要加一個use

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

int n, k;
int a[15], use[15];

void dfs(int pos) {
    if(pos ==k+1) {
        for(int i = 1; i <= k; i++)
            cout << a[i] << ' ';
        cout << endl;

        return;
    }

    for(int i = 1; i <= n; i++)
        if(!use[i]) { // 如果第 i 個元素沒有被使用
            use[i] = 1; // 將其標記為被使用
            a[pos]=i; // 更新 a[pos]
            dfs(pos+1); // 列舉下一個
            use[i] =0; // 將其標記為未被使用
        }
}

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

    dfs(1);

    return 0;
}

如果是變種的話,則需要將程式碼中i的起始數設定為上一個數+1,這樣可以保證不會2,1/1,2出現

下面看主角:P1036題目傳送門

[NOIP2002 普及組] 選數

題目描述

已知 $n$ 個整數 $x_1,x_2,\cdots,x_n$,以及 $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 \le n \le 20$,$k<n$)。

第二行 $n$ 個整數,分別為 $x_1,x_2,\cdots,x_n$($1 \le x_i \le 5\times 10^6$)。

輸出格式

輸出一個整數,表示種類數。

樣例 #1

樣例輸入 #1

4 3
3 7 12 19

樣例輸出 #1

1

提示

【題目來源】

NOIP 2002 普及組第二題

解析

分析問題可以得出,他就是我們剛剛提到的變種,交換順序不算
1.構建選數函式(dfs)
2.判斷質數
3.輸出結果
下面是程式碼

#include <bits/stdc++.h>

using namespace std;
int n,r;
int a[1005];
int in[1005];
int ans;
bool prime(int x){//判斷質數
    if(x<=1)return false;
    if(x==2) return true;
    for(int i=2;i<=sqrt(x);i++){//增加效率
        if(x%i==0)return false;
    }
    return true;

}
void dfs(int k){//選數
if(k>r){
    int cnt=0;
    for(int i=1;i<=r;i++){
        cnt+=in[a[i]];

    }
    //cout<<cnt<<endl;
   if(prime(cnt)) ans++;
   return;
}
for(int i=a[k-1]+1;i<=n;i++){
    a[k]=i;
    dfs(k+1);
}
}
int main()
{
   cin>>n>>r;
   for(int i=1;i<=n;i++){
       cin>>in[i];
   }
   dfs(1);//從第一個數開始
   cout<<ans;
   return 0;
}
//https://www.luogu.com.cn/problem/P1036

謝謝大家

相關文章