洛谷P1631 序列合并

序列合并,将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;
		}
	}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值