1. 程式人生 > 其它 >cf264 C. Choosing Balls(dp)

cf264 C. Choosing Balls(dp)

題意:

n個球排成一列,每個球有值 \(v_i\) 和顏色 \(c_i\)。可任取子序列,對子序列中的某個球,若它不是子序列首且與子序列中的上一個球同色,則價值為 \(av_i\);若它是子序列首或者與子序列中的上一個球異色,則價值為 \(bv_i\)。子序列的價值定義為子序列中所有球的價值和。

現有q次詢問,每次給定 \(a,b\)。對每個詢問,找一價值最大的子序列並輸出價值。

\(1\le n \le 1e5, 1\le q \le 500, |v_i|\le 1e5,1\le c_i \le n, |a|\le 1e5,|b| \le 1e5\)

思路:

\(f(c)\) 表示最後一個球的顏色為 \(c\)

的子序列的最大價值。對每個詢問都要重新做一遍dp。

每個球 \(i\) 可以接到前面的末尾同色的子列後:\(f(c_i)=f(c_i) + av_i\)

也可以接到前面的最大的末尾異色子列後:\(f(c_i)=\mathop{max}\limits _{c_j\neq c_i}\{f(c_j)\} + bv_i\)

其中最大的 \(f\) 的值是遞增的,可以邊迴圈邊維護。為了異色,要維護最大顏色和次大顏色。

複雜度 \(O(nq)\)

int a, b, mi1 = 0, mi2 = 0; cin >> a >> b;

memset(f, 0xcf, sizeof f); f[0] = 0;

for(int i = 1; i <= n; i++)
{
    int t = (c[i] != c[mi1]) ? mi1 : mi2; //異色
    
    f[c[i]] = max({f[c[i]], f[c[i]] + (ll)v[i] * a, f[c[t]] + (ll)v[i] * b}); //dp

    if(f[c[mi1]] < f[c[mi2]]) swap(mi1, mi2); //維護最大和次大
    if(c[i] == c[mi1] || c[i] == c[mi2]) continue;
    if(f[c[i]] > f[c[mi1]]) mi2 = mi1, mi1 = i;
    else if(f[c[i]] > f[c[mi2]]) mi2 = i;
}