序列合并,将a1b1入队,然后从当前节点aibj开始扩展,扩展到ai+1bj和aibj+1,扔到优先队列里,节点可能被重复扩展,设置数字t,存储当点ai已经扩展到哪个节点。交上去之后40分。
#include<iostream>
#include<set>
#include<stdio.h>
#include<queue>
using namespace std;
const int N=1E5+10;
struct node{
int x,y,val;
bool operator <( node b)const{
return val>b.val;
}
node(int i,int j,int k):x(i),y(j),val(k){}
};
priority_queue<node>pq;
int a[N],b[N],t[N],n;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",a+i);
for(int i=1;i<=n;i++)
scanf("%d",b+i);
a[n+1]=b[n+1]=2e9+10;
pq.push({1,1,a[1]+b[1]});
t[1]=1;
for(int i=1;i<=n;i++){
node tmp=pq.top();
pq.pop();
printf("%d ",tmp.val);
if(t[tmp.x]<n&&t[tmp.x]<tmp.y+1)
pq.push({tmp.x,tmp.y+1,a[tmp.x]+b[tmp.y+1]}),t[tmp.x]=tmp.y+1;
if(t[tmp.x+1]<n&&t[tmp.x+1]<tmp.y)
pq.push({tmp.x+1,tmp.y,a[tmp.x+1]+b[tmp.y]}),t[tmp.x+1]=tmp.y;
}
}
自己想不明白原因,但应该是扩展或去重写错了,修改成set判重,过了
for(int i=1; i<=n; i++) {
node tmp=pq.top();
pq.pop();
printf("%d ",tmp.val);
if(tmp.y+1>t[tmp.x]) {
for(int j=t[tmp.x]+1; j<=n&&j<=tmp.y+1; j++)
pq.push({tmp.x,j,a[tmp.x]+b[j]});
t[tmp.x]=tmp.y+1;
}
if(tmp.y>t[tmp.x+1]) {
for(int j=t[tmp.x+1]+1; tmp.x+1<=n&&j<=n&&j<=tmp.y; j++)
pq.push({tmp.x+1,j,a[tmp.x+1]+b[j]});
t[tmp.x+1]=tmp.y;
}
}
前面的代码错在哪儿呢,查找数据发现扩展的时候有些节点漏了,比如扩展i+1,j时候,可能前面的i+1,j-1还没有扩展呢.修改后,过了
for(int i=1; i<=n; i++) {
node tmp=pq.top();
pq.pop();
printf("%d ",tmp.val);
if(tmp.y+1>t[tmp.x]) {
for(int j=t[tmp.x]+1; j<=n&&j<=tmp.y+1; j++)
pq.push({tmp.x,j,a[tmp.x]+b[j]});
t[tmp.x]=tmp.y+1;
}
if(tmp.y>t[tmp.x+1]) {
for(int j=t[tmp.x+1]+1; tmp.x+1<=n&&j<=n&&j<=tmp.y; j++)
pq.push({tmp.x+1,j,a[tmp.x+1]+b[j]});
t[tmp.x+1]=tmp.y;
}
}