CodeForces 30 C.Shooting Gallery（概率DP）

最新推荐文章于 2020-02-08 21:18:53 发布

v5zsq

最新推荐文章于 2020-02-08 21:18:53 发布

阅读量372

点赞数

分类专栏： Code Forces 概率DP

本文链接：https://blog.csdn.net/V5ZSQ/article/details/79762301

版权

Code Forces 同时被 2 个专栏收录

606 篇文章 6 订阅

订阅专栏

概率DP

23 篇文章 0 订阅

订阅专栏

Description

在一个垂直的二维平板给出 $n$ 个目标的坐标，每次瞄完一个目标后实现在平板上移动的速度为 $1$ ，瞄准第 $i$ 个目标后射中的概率为 $p_i$ ，且第 $i$ 个目标只在第 $t_i$ 秒出现，也就是说只能在第 $t_i$ 秒瞄准该目标并射击，问射中目标数的期望值最大为多少

Input

第一行一整数 $n$ 表示目标个数，之后 $n$ 行每行输入四个数 $x_i,y_i,t_i,p_i$ 分别表示第 $i$ 个目标的坐标，出现的时刻以及射中的概率 $(1\le n\le 1000,-10^4\le x_i,y_i\le 10^4,0\le t_i\le 10^0,0\le p_i\le 1)$

Output

输出射中目标数量期望值的最大值

Sample Input

1
0 0 0 0.5

Sample Output

0.5000000000

Solution

把所有目标按出现时刻排序，如果打完第 $i$ 个目标后要打第 $j$ 个目标，那么需要满足 $dis(i,j)\le t_j-t_i,j>i$ ，用 $dp[i]$ 表示最后一个打的是第 $i$ 个目标的最大期望值，那么 $dp[1]=p_1$ ,且有转移 $dp[j]=max(dp[i]+p_j),dis(i,j)\le t_j-t_i,j>i$ ， $max(dp[i],1\le i\le n)$ 即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
int n;
struct node
{
    int x,y,t;
    double p;
    bool operator<(const node &b)const
    {
        return t<b.t;
    }
}a[maxn];
double dp[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d%d%d%lf",&a[i].x,&a[i].y,&a[i].t,&a[i].p);
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)dp[i]=a[i].p;
    double ans=dp[1];
    for(int i=2;i<=n;i++)
    {   
        for(int j=i-1;j>=1;j--)
        {
            ll D=(ll)(a[i].x-a[j].x)*(a[i].x-a[j].x)+(ll)(a[i].y-a[j].y)*(a[i].y-a[j].y);
            ll T=(ll)(a[i].t-a[j].t)*(a[i].t-a[j].t);
            if(D<=T)dp[i]=max(dp[i],dp[j]+a[i].p);
        }
        ans=max(ans,dp[i]); 
    } 
    printf("%.9f\n",ans);
    return 0;
}