String of Colorful Beads

Eva would like to buy a string of beads with no repeated colors so she went to a small shop of which the owner had a very long string of beads. However the owner would only like to cut one piece at a time for his customer. With as many as ten thousand beads in the string, Eva needs your help to tell her how to obtain the longest piece of string that contains beads with all different colors. And more, each kind of these beads has a different value. If there are more than one way to get the longest piece, Eva would like to take the most valuable one. It is guaranteed that such a solution is unique.

Input Specification:

Each input file contains one test case. Each case first gives in a line a positive integer N (≤ 10,000) which is the length of the original string in the shop. Then N positive numbers (≤ 100) are given in the next line, which are the values of the beads -- here we assume that the types of the beads are numbered from 1 to N, and the i-th number is the value of the i-th type of beads. Finally the last line describes the original string by giving the types of the beads from left to right. All the numbers in a line are separated by spaces.

Output Specification:

For each test case, print in a line the total value of the piece of string that Eva wants, together with the beginning and the ending indices of the piece (start from 0). All the numbers must be separated by a space and there must be no extra spaces at the beginning or the end of the line.

Sample Input:

8
18 20 2 97 23 12 8 5
3 3 5 8 1 5 2 1

 

Sample Output:

66 3 6

#include<bits/stdc++.h>
using namespace std;
int main()
{
//    freopen("data.txt","r",stdin);
    int n,k;
    scanf("%d",&n);
    vector<int> v,vl;
    vector<int> vi(n+1,-1);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&k);
        v.emplace_back(k);
    }
    int vls=0,b=-1,e=-1,s=0,mb=0,ms=0;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&k);
        vls+=v[k-1];
        vl.emplace_back(vls);
        if(vi[k]>=mb)
        {
            if(s<ms)
            {
                b=mb;
                e=i-1;
                s=ms;
            }
            ms=vls-vl[vi[k]];
            mb=vi[k]+1;
        }
        else
        ms+=v[k-1];
        vi[k]=i;
    }
    if(s<ms)
    {
        b=mb;
        e=n-1;
        s=ms;
    }
    printf("%d %d %d",s,b,e);
    return 0;
}