AtCoder Grand Contest 015 E - Mr.Aoki Incubator dp

最新推荐文章于 2020-08-30 10:00:31 发布

_beginend

最新推荐文章于 2020-08-30 10:00:31 发布

阅读量336

点赞数

分类专栏：动态规划

本文链接：https://blog.csdn.net/qq_33229466/article/details/79238105

版权

动态规划专栏收录该内容

127 篇文章 0 订阅

订阅专栏

题意

数轴上有n个点，每个点有一个初始坐标和速度。一开始可以选择把若干个点染红。开始运动后，若红点和一个未染色点相撞，则该点会被染成红色。问有多少种染色方案使得在无限久后所有点都会变成红色。
n<=200000

分析

一开始想的是按坐标排序然后讨论，发现不好搞。
实际上可以按速度排序，设 $lef[i]$ 表示最小的 $j$ 满足 $x[j]>=x[i]$ ,设 $rig[i]$ 表示最大的 $j$ 满足 $x[j]<=x[i]$ 。找下规律不难发现当染红点 $i$ 后，区间 $[lef[i],rig[i]]$ 中的点都会被染红。于是问题就变成了有 n <script type="math/tex" id="MathJax-Element-27">n</script>个区间，有多少种选择区间的方案使得选出来的区间并集为全集。
直接dp即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int MOD=1000000007;
const int N=200005;

int n,f[N],s[N],stack[N],lef[N],rig[N];
struct data{int x,v;}a[N];
struct Interval{int l,r;}itv[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

bool cmp(data a,data b)
{
    return a.v<b.v;
}

bool cmp1(Interval a,Interval b)
{
    return a.r<b.r||a.r==b.r&&a.l<b.l;
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i].x=read(),a[i].v=read();
    sort(a+1,a+n+1,cmp);
    int top=0;
    for (int i=1;i<=n;i++)
    {
        if (!top||a[i].x>a[stack[top]].x) stack[++top]=i;
        int l=1,r=top;
        while (l<=r)
        {
            int mid=(l+r)/2;
            if (a[stack[mid]].x>=a[i].x) r=mid-1;
            else l=mid+1;
        }
        lef[i]=stack[r+1];
    }
    top=0;
    for (int i=n;i>=1;i--)
    {
        if (!top||a[i].x<a[stack[top]].x) stack[++top]=i;
        int l=1,r=top;
        while (l<=r)
        {
            int mid=(l+r)/2;
            if (a[stack[mid]].x<=a[i].x) r=mid-1;
            else l=mid+1;
        }
        rig[i]=stack[r+1];
    }
    for (int i=1;i<=n;i++) itv[i].l=lef[i],itv[i].r=rig[i];
    sort(itv+1,itv+n+1,cmp1);
    int now=1,ans=0;
    for (int i=1;i<=n;i++)
    {
        while (itv[now].r<itv[i].l-1) now++;
        f[i]=(s[i-1]+MOD-s[now-1])%MOD;
        if (itv[i].l==1) f[i]++;
        if (itv[i].r==n) (ans+=f[i])%=MOD;
        s[i]=(s[i-1]+f[i])%MOD;
    }
    printf("%d",ans);
    return 0;
}