1. 程式人生 > >2018 icpc南京網路賽G(線段樹)

2018 icpc南京網路賽G(線段樹)

思路:因為只有1e5的資料。我們可以模擬一下所有情況全算出來,把答案存下來即可。但會一遍一遍的遍歷超時,我們可以優化遍歷的方式,因為只有1e5的房間,我們每次找的房間都是小於當前新燈泡數且最前面的房間。這個查詢用一顆線段樹既可以解決。

我們每次找到一個滿足的房間就讓當前燈泡是減去這個房間的燈泡數,然後讓當前房間的燈泡數設成無窮大,更新一下線段樹即可。

#include <iostream>
#include <string>
#include <string.h>
#include <algorithm>
#include <vector>
#define MAXN 1005
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
int a[100005];
struct node{
	int l,r,val;
}c[400005];
int cnt,n,m;
void build(int l,int r,int rt)
{
	c[rt].l=l;c[rt].r=r;
	if(l==r)
	{
		c[rt].val=a[l];
		return ;
	}
	int mid=(l+r)>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	c[rt].val=min(c[rt<<1].val,c[rt<<1|1].val);
}
int now;
int query(int l,int r,int rt,int val)
{
	if(l==r)
	{
		cnt++;
		now-=c[rt].val;c[rt].val=INF;
		return c[rt].l;
	}
	int mid=(l+r)>>1;
	int ans=0;
	if(c[rt<<1].val<=val)
	{
		ans=query(l,mid,rt<<1,val);
	}
	else if(c[rt<<1|1].val<=val)
	{
		ans=query(mid+1,r,rt<<1|1,val);
	}
	c[rt].val=min(c[rt<<1].val,c[rt<<1|1].val);
	return ans;
}
int ans1[1000007],ans2[1000007];
int main()
{
	cnt=0;
	int t;
	scanf("%d%d",&n,&m);
	now=m;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	build(1,n,1);
	int q;scanf("%d",&q);
	int i=1;
	int qop=0x3f3f3f3f;
	while(cnt!=n&&i<=100005)
	{
		int op=0;
		int pos=query(1,n,1,now);
		while(pos!=0)
		{
			op++;
			pos=query(1,n,1,now);
		}
		if(cnt==n) qop=i;
		ans1[i]=op+ans1[i-1];ans2[i]=now;
		i++;now+=m;
	}
	while(q--)
	{
		int x;scanf("%d",&x);
		if(x>qop)
		printf("%d %d\n",ans1[qop],ans2[qop]);
		else
		printf("%d %d\n",ans1[x],ans2[x]);
	}
    return 0;
}