題目描述

有兩個長度都是N的序列A和B，在A和B中各取一個數相加可以得到N^2N2個和，求這N^2N2個和中最小的N個。

輸入輸出格式

輸入格式：

?

第一行一個正整數N；

第二行N個整數A_iAi?, 滿足A_i\le A_{i+1}Ai?≤Ai+1?且A_i\le 10^9Ai?≤109;

第三行N個整數B_iBi?, 滿足B_i\le B_{i+1}Bi?≤Bi+1?且B_i\le 10^9Bi?≤109.

【數據規模】

對于50%的數據中，滿足1<=N<=1000；

對于100%的數據中，滿足1<=N<=100000。

?

輸出格式：

?

輸出僅一行，包含N個整數，從小到大輸出這N個最小的和，相鄰數字之間用空格隔開。

?

輸入輸出樣例

輸入樣例#1：
3 2 6 6 1 4 8 輸出樣例#1：
3 6 7

?

題解

• 其實就是每次找最小的兩個，但是題目是每個數可以用多次，如何不重復的去做才是終點
• 首先，對于A序列的第一個數一定是要選的，那么就可以先把B數列中所有數與它的和加入優先隊列中，并記錄下當前與A序列第一個數相加，可以用P數組存
• 那么每次就輸出隊首，這個沒問題吧
• 然后將對首的A數列變后一個，將之前記錄P數組+1，再加上自己，再加入隊列里

代碼

1 #include <cstdio> 2 #include <queue> 3 #include <vector> 4 #define N 100010 5 using namespace std; 6 priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > >Q; 7 int n,k,a[N],b[N],p[N]; 8 int main() 9 { 10 scanf("%d",&n); 11 for (int i=1;i<=n;i++) scanf("%d",&a[i]); 12 for (int i=1;i<=n;i++) scanf("%d",&b[i]),p[i]=1,Q.push(make_pair(a[1]+b[i],i)); 13 while (n--) printf("%d ",Q.top().first),k=Q.top().second,Q.pop(),Q.push(make_pair(a[++p[k]]+b[k],k)); 14 }

?

轉載于:https://www.cnblogs.com/Comfortable/p/10327991.html

總結

以上是生活随笔為你收集整理的[优先队列] 洛谷 P1631 序列合并的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。