Codeforces Round #377 (Div. 2) D. Exams (二分)

D. Exams
time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output

Vasiliy has an exam period which will continue for n days. He has to pass exams on m subjects. Subjects are numbered from 1 to m.

About every day we know exam for which one of m subjects can be passed on that day. Perhaps, some day you can't pass any exam. It is not allowed to pass more than one exam on any day. 

On each day Vasiliy can either pass the exam of that day (it takes the whole day) or prepare all day for some exam or have a rest. 

About each subject Vasiliy know a number ai — the number of days he should prepare to pass the exam number i. Vasiliy can switch subjects while preparing for exams, it is not necessary to prepare continuously during ai days for the exam number i. He can mix the order of preparation for exams in any way.

Your task is to determine the minimum number of days in which Vasiliy can pass all exams, or determine that it is impossible. Each exam should be passed exactly one time. 

Input

The first line contains two integers n and m (1 ≤ n, m ≤ 105) — the number of days in the exam period and the number of subjects. 

The second line contains n integers d1, d2, ..., dn (0 ≤ di ≤ m), where di is the number of subject, the exam of which can be passed on the day number i. If di equals 0, it is not allowed to pass any exams on the day number i

The third line contains m positive integers a1, a2, ..., am (1 ≤ ai ≤ 105), where ai is the number of days that are needed to prepare before passing the exam on the subject i.

Output

Print one integer — the minimum number of days in which Vasiliy can pass all exams. If it is impossible, print -1.

Examples
input
7 2
0 1 0 2 1 0 2
2 1
output
5
input
10 3
0 0 1 2 3 0 2 0 1 2
1 1 4
output
9
input
5 1
1 1 1 1 1
5
output
-1
Note

In the first example Vasiliy can behave as follows. On the first and the second day he can prepare for the exam number 1 and pass it on the fifth day, prepare for the exam number 2 on the third day and pass it on the fourth day.

In the second example Vasiliy should prepare for the exam number 3 during the first four days and pass it on the fifth day. Then on the sixth day he should prepare for the exam number 2 and then pass it on the seventh day. After that he needs to prepare for the exam number 1 on the eighth day and pass it on the ninth day. 

In the third example Vasiliy can't pass the only exam because he hasn't anough time to prepare for it. 

题意:
给你n天,m门课,每天你可以选择考一门或者复习一门。每一天所能够选择的考试的那一门课序号会给出,复习不限。
问你最短能在几天内完成所有考试,若无法在n天内完成,输出-1。
分析:
很有趣的一道题。开始没想到二分解法,只想着先判断n天能否完成,若能完成再继续考虑是否能得出最少天数。后来发现别人的做法是直接二分天数,直接判断该天能否完成考试,从而求出最小天数,或输出-1.

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <string>
#include <vector>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
using namespace std;
#define INF 0x3f3f3f3f
const long long N=1e5+5;
const long long mod=1e9;
const double PI=acos(-1.0);
typedef long long ll;

int a[N];
int d[N];
int pre[N];
int n,m;
bool judge(int day){
//    cout<<day<<endl;
    int res=0;
    int flag=0;
    memset(pre, 0, sizeof(pre));
    for (int i=day-1; i>=0; i--) {
        if (d[i]) {
            if (!pre[d[i]]) {
                res+=a[d[i]];
                pre[d[i]]=1;
                flag++;
            } else if(res!=0) {
                res--;
            }
        } else if(res!=0) {
            res--;
        }
    }
    if (flag!=m||res) {
        return false;
    }
    return true;
}

int main() {
    while (cin>>n>>m) {
        for (int i=0; i<n; i++) {
            scanf("%d",&d[i]);
        }
        for (int i=0; i<m; i++) {
            scanf("%d",&a[i+1]);
        }
        
        int l=1,r=n,mid;
        while (l<r) {
            mid=l+(r-l)/2;
            if (judge(mid)) {
                r=mid;
            } else {
                l=mid+1;
            }
        }
        if (judge(l)) {
            cout<<l<<endl;
        } else {
            cout<<"-1\n";
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值