[Atcoder Yahoo Contest 2019]D.Ears(动态规划)

Score: 600 600 600 points

题面

传送门
翻译有时间再补…

题解

体验感极差,考试的时候手残把1打成了2Debug了半个小时,害的F题都没做。
先将题目转换一下:给你一条链,顺次连接着 n + 1 n+1 n+1个点,每条边有一个边权 A i A_i Ai,在有一个人从任意一个起点 P P P出发,设他经过第 i i i条边的次数 D i D_i Di,求 ∑ i = 1 n ∣ A i + D i ∣ \sum^{n}_{i=1}|A_i+D_i| i=1nAi+Di的最小值。
首先需要证明一个结论:
A_i为奇数,等价于 A i = 1 A_i=1 Ai=1
A_i为大于0的偶数,等价于 A i = 2 A_i=2 Ai=2
证明很简单,可以发现走到边的同一侧的 D i D_i Di奇偶性相同,由于你经过这条边的时候可以任意来回跳动偶数次,那么答案一定不会变的更劣。
考虑一个人从 P P P点出发,有两种方案:
1.先往左边走,然后回到 P P P点,再往右边走,不再回来。
2.先往右边走,然后回到 P P P点,再往左边走,不再回来。
这很像一个dp。
定义状态 L 1 i , L 2 i , R 1 i , R 2 i L1_i,L2_i,R1_i,R2_i L1i,L2i,R1i,R2i
分别代表从 i i i点向左走需要回来的最小代价,从 i i i点向左走不需要回来的最小代价,从 i i i点向右走需要回来的最小代价,从 i i i点向右走不需要回来的最小代价。

L 1 [ i ] = { min ⁡ ( L 1 i − 1 + 2 , l s u m i ) ( A i = = 0 ) min ⁡ ( L 1 i − 1 + 1 , l s u m i ) ( A i = = 1 ) min ⁡ ( L 1 i − 1 , l s u m i ) ( A i = = 2 ) L1[i]=\begin{cases} \min(L1_{i-1}+2,lsum_i)(A_i==0)\\ \min(L1_{i-1}+1,lsum_i)(A_i==1)\\ \min(L1_{i-1},lsum_i)(A_i==2)\\ \end{cases} L1[i]=min(L1i1+2,lsumi)(Ai==0)min(L1i1+1,lsumi)(Ai==1)min(L1i1,lsumi)(Ai==2)
L 2 [ i ] = { min ⁡ ( L 1 i , m i n ( L 2 i − 1 + 1 , l s u m i ) ) ( A i = = 0 ) min ⁡ ( L 1 i , m i n ( L 2 i − 1 , l s u m i ) ) ( A i = = 1 ) min ⁡ ( L 1 i , m i n ( L 2 i − 1 + 1 , l s u m i ) ) ( A i = = 2 ) L2[i]=\begin{cases}\min(L1_i,min(L2_{i-1}+1,lsum_i))(A_i==0)\\ \min(L1_i,min(L2_{i-1},lsum_i))(A_i==1)\\ \min(L1_i,min(L2_{i-1}+1,lsum_i))(A_i==2)\\ \end{cases} L2[i]=min(L1i,min(L2i1+1,lsumi))(Ai==0)min(L1i,min(L2i1,lsumi))(Ai==1)min(L1i,min(L2i1+1,lsumi))(Ai==2)
R同理。
需要注意的是在走路的过程中可以随时不走后面的路调头,代价是前面/后面 i i i个数的 A i A_i Ai之和。
答案就是 m i n ( L 1 [ i ] + R 2 [ i ] , L 2 [ i ] + R 1 [ i ] ) min(L1[i]+R2[i],L2[i]+R1[i]) min(L1[i]+R2[i],L2[i]+R1[i])
时间复杂度: O ( n ) O(n) O(n)
[题解好像还有一种神仙写法,我菜爆了…]

#include
#include
#include
#include
#include
#include
using namespace std;
#define MAXN 200000
#define LL long long
#define DB double
#define FR first
#define SE second
int a[MAXN+5];
LL ans;
LL L1[MAXN],L2[MAXN+5],R1[MAXN+5],R2[MAXN+5];
LL lsum[MAXN+5],rsum[MAXN+5];
int n;
int get_num(int p)
{
	if(p==0)return 0;
	if(p%2==1)return 1;
	return 2;
}
int main()
{
	ans=1000000000;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    lsum[i]=lsum[i-1]+a[i];
    for(int i=n-1;i>=0;i--)
    rsum[i]=rsum[i+1]+a[i+1];
    for(int i=1;i<=n;i++)a[i]=get_num(a[i]);
    for(int i=1;i<=n;i++)
    {
        if(a[i]==0)L1[i]=min(L1[i-1]+2,lsum[i]);
        else if(a[i]==1)L1[i]=min(L1[i-1]+1,lsum[i]);
        else L1[i]=min(L1[i-1],lsum[i]);
        if(a[i]==0)L2[i]=min(L1[i],min(L2[i-1]+1,lsum[i]));
        else if(a[i]==1)L2[i]=min(L1[i],min(L2[i-1],lsum[i]));
        else L2[i]=min(L1[i],min(L2[i-1]+1,lsum[i]));
    }
    for(int i=n-1;i>=0;i--)
    {
        if(a[i+1]==0)R1[i]=min(R1[i+1]+2,rsum[i]);
        else if(a[i+1]==1)R1[i]=min(R1[i+1]+1,rsum[i]);
        else R1[i]=min(R1[i+1],rsum[i]);
        if(a[i+1]==0)R2[i]=min(R1[i],min(R2[i+1]+1,rsum[i]));
        else if(a[i+1]==1)R2[i]=min(R1[i],min(R2[i+1],rsum[i]));
        else R2[i]=min(R1[i],min(R2[i+1]+1,rsum[i]));
    }
    for(int i=0;i<=n;i++)
    {
    	//printf("%d %lld %lld\n",i,L1[i]+R2[i],L2[i]+R1[i]);
        ans=min(ans,min(L1[i]+R2[i],L2[i]+R1[i]));
    }
    printf("%lld\n",ans);
}

你可能感兴趣的:([Atcoder Yahoo Contest 2019]D.Ears(动态规划))