郵局(深搜+剪枝)

題目如下:

問題描述
  C村住著n戶村民,由于交通閉塞,C村的村民只能通過信件與外界交流。為了方便村民們發信,C村打算在C村建設k個郵局,這樣每戶村民可以去離自己家最近的郵局發信。

  現在給出了m個備選的郵局,請從中選出k個來,使得村民到自己家最近的郵局的距離和最小。其中兩點之間的距離定義為兩點之間的直線距離。
輸入格式
  輸入的第一行包含三個整數n, m, k,分別表示村民的戶數、備選的郵局數和要建的郵局數。
  接下來n行,每行兩個整數x, y,依次表示每戶村民家的坐標。
  接下來m行,每行包含兩個整數x, y,依次表示每個備選郵局的坐標。
  在輸入中,村民和村民、村民和郵局、郵局和郵局的坐標可能相同,但你應把它們看成不同的村民或郵局。
輸出格式
  輸出一行,包含k個整數,從小到大依次表示你選擇的備選郵局編號。(備選郵局按輸入順序由1到m編號)
樣例輸入
5 4 2
0 0
2 0
3 1
3 3
1 1
0 1
1 0
2 1
3 2
樣例輸出
2 4
數據規模和約定
  對于30%的數據,1<=n<=10,1<=m<=10,1<=k<=5;
  對于60%的數據,1<=m<=20;
  對于100%的數據,1<=n<=50,1<=m<=25,1<=k<=10。

這道題使用的方法是剪枝+深搜。
解題步驟在于:
先設定一個郵局是已確定的,得出所有的村民到該郵局的距離。
再逐個進行判斷未確定的郵局,如果該郵局到所有村民家的距離都大于當前最小值,那么該郵局被剪枝,即被放棄。
直到找遍所有的郵局且郵局數目為當前數目。
使用方法深搜則在于,每當確定一個郵局,那么下一個參數的傳遞值則加一,來標記已確定郵局數目。
其中需要注意到的是,從大到小首先確定一個數,然后從大到小依次判斷是否可留,如果是,那么替換當前已確定最小數,否則,被剪枝。繼續迭代。

#include<iostream>  //郵局  
#include<stdlib.h>
#include<math.h>
using namespace std;
int n, m, k, j, c[55][2], y[27][2], d[12], f1, f2, f[55] = { 0 };
float yc[27][55], s = 1000000000;
int dfs(int t, int i, int o[12], float w[55], float sum)
{
    if (i <= m + 1)//如果還沒有遍歷完所有的郵局
    {
        if (t == k)//如果已經確定的郵局數已足夠
        {
            if (sum<s)
            {
                s = sum;//s是最后的最小距離總值
                for (j = 0; j<k; j++)
                    d[j] = o[j];//將數組o的k個值存入數組d
            }
        }
        else if (i <= m&&t<k)//還沒有確定所有郵局數
        {
            float ww[55];
            for (j = 1; j <= n; j++)
                ww[j] = w[j];
            dfs(t, i + 1, o, w, sum); f1 = 1, f2 = 0;//下一個郵局,初始化兩個標記用的變量f1,f2
            if (!f[i])//f[i]==0,還沒有被剪掉
            {
                o[t] = i;//第t個已確定郵局是i
                if (t>0)//ww初始化已經
                {
                    f2 = 1;
                    for (j = 1; j <= n; j++)
                    {
                        if (ww[j]>yc[i][j])//如果郵局到村民家的距離小于當前最小
                        {
                            sum = sum - ww[j] + yc[i][j];//更新
                            ww[j] = yc[i][j];
                            f1 = 0;//變化,不剪掉當前郵局
                        }
                    }
                }
                else//還沒有初始化
                {
                    for (j = 1; j <= n; j++)
                    {
                        sum += yc[i][j];
                        ww[j] = w[j] = yc[i][j];//初始化最小值就是當前值
                    }
                } 
                if (f1&&f2)//已經有過ww初始化且需要剪掉當前的郵局,ww如果未初始化那么一定不能剪掉
                {
                    f[i] = 1;//經過處理,已經被剪掉
                    dfs(t, i + 1, o, w, sum);//下一次迭代t不增加
                }
                else
                    dfs(t + 1, i + 1, o, ww, sum);//下一次迭代
            }
        }
    }
}
int main()
{
    int i, j, o[12];
    float w[55], ww[55];
    cin >> n >> m >> k;
    for (i = 1; i <= n; i++)
        cin >> c[i][0] >> c[i][1];
    for (i = 1; i <= m; i++)
    {
        cin >> y[i][0] >> y[i][1];
        for (j = 1; j <= n; j++)
            yc[i][j] = sqrt((c[j][0] - y[i][0])*(c[j][0] - y[i][0]) + (c[j][1] - y[i][1])*(c[j][1] - y[i][1]));
    }//yc[i][j]代表第i個郵局到第j個村民家的距離
    dfs(0, 1, o, w, 0);
    for (i = 0; i<k; i++)
        cout << d[i] << " ";
    return 0;
}
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,983評論 6 537
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,772評論 3 422
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,947評論 0 381
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,201評論 1 315
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,960評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,350評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,406評論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,549評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,104評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,914評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,089評論 1 371
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,647評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,340評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,753評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,007評論 1 289
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,834評論 3 395
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,106評論 2 375

推薦閱讀更多精彩內容

  • 決策樹理論在決策樹理論中,有這樣一句話,“用較少的東西,照樣可以做很好的事情。越是小的決策樹,越優于大的決策樹”。...
    制杖灶灶閱讀 5,891評論 0 25
  • 機器學習 經驗 數據 數據中產生模型model 的算法 學習算法 learning algorithm 數據集 d...
    時待吾閱讀 4,007評論 0 3
  • 一.樸素貝葉斯 1.分類理論 樸素貝葉斯是一種基于貝葉斯定理和特征條件獨立性假設的多分類的機器學習方法,所...
    wlj1107閱讀 3,116評論 0 5
  • 也許每個人心里想法都不一樣,但是我希望有人看到這句話,你們可以好好聽對方講完自己內心的想法。然后再來否定他的想法
    叔懶閱讀 126評論 0 0
  • @楊磊老師? “禮樂皆得謂之有德。德者,得也。”《禮記·樂記》篇。 《禮記·中庸》:“故大德必得其位,必得其祿,必...
    楊仁珮001閱讀 496評論 0 0