Codeforces 264 C. Choosing Balls


DP,应为前面一个球有颜色相同和不同两种情况,维护两个颜色不同的球的最值,就可以转移了.

C. Choosing Balls
time limit per test
5 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

There are n balls. They are arranged in a row. Each ball has a color (for convenience an integer) and an integer value. The color of the i-th ball is ci and the value of the i-th ball is vi.

Squirrel Liss chooses some balls and makes a new sequence without changing the relative order of the balls. She wants to maximize the value of this sequence.

The value of the sequence is defined as the sum of following values for each ball (where a and b are given constants):

  • If the ball is not in the beginning of the sequence and the color of the ball is same as previous ball's color, add (the value of the ball)  × a.
  • Otherwise, add (the value of the ball)  ×  b.

You are given q queries. Each query contains two integers ai and bi. For each query find the maximal value of the sequence she can make when a = ai and b = bi.

Note that the new sequence can be empty, and the value of an empty sequence is defined as zero.

Input

The first line contains two integers n and q (1 ≤ n ≤ 105; 1 ≤ q ≤ 500). The second line contains n integers: v1, v2, ..., vn (|vi| ≤ 105). The third line contains n integers: c1, c2, ..., cn (1 ≤ ci ≤ n).

The following q lines contain the values of the constants a and b for queries. The i-th of these lines contains two integers ai and bi(|ai|, |bi| ≤ 105).

In each line integers are separated by single spaces.

Output

For each query, output a line containing an integer — the answer to the query. The i-th line contains the answer to the i-th query in the input order.

Please, do not write the %lld specifier to read or write 64-bit integers in С++. It is preferred to use the cincout streams or the %I64dspecifier.

Sample test(s)
input
6 3
1 -2 3 4 0 -1
1 2 1 2 1 1
5 1
-2 1
1 0
output
20
9
4
input
4 1
-3 6 -1 2
1 2 3 1
1 -1
output
5
Note

In the first example, to achieve the maximal value:

  • In the first query, you should select 1st, 3rd, and 4th ball.
  • In the second query, you should select 3rd, 4th, 5th and 6th ball.
  • In the third query, you should select 2nd and 4th ball.

Note that there may be other ways to achieve the maximal value.



#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long int LL;

const LL INF=(1LL<<60);

int n,q;
LL c[100100],v[100100];
LL dp[100100],x,y;

int main()
{
    cin>>n>>q;
    for(int i=1;i<=n;i++) cin>>v[i];
    for(int i=1;i<=n;i++) cin>>c[i];

    while(q--)
    {
        cin>>x>>y;
        fill(dp,dp+n+10,-INF);
        int id1=0,id2=0;
        LL ans=0;

        for(int i=1;i<=n;i++)
        {
            dp[c[i]]=max(dp[c[i]],max(dp[c[i]]+v[i]*x,v[i]*y));
            if(id1!=c[i])
                dp[c[i]]=max(dp[c[i]],dp[id1]+v[i]*y);
            else
                dp[c[i]]=max(dp[c[i]],dp[id2]+v[i]*y);
            if(id1!=c[i])
            {
                if(dp[c[i]]>dp[id1])
                {
                    id2=id1; id1=c[i];
                }
                else if(dp[c[i]]>dp[id2])
                {
                    id2=c[i];
                }
            }
            ans=max(ans,dp[c[i]]);
        }

        cout<<ans<<endl;
    }

    return 0;
}




你可能感兴趣的:(Codeforces 264 C. Choosing Balls)