Visiting Peking University(水)

Visiting Peking University

时间限制: 1000ms

单点时限: 1000ms

内存限制: 256MB

描述

Ming is going to travel for n days and the date of these days can be represented by n integers: 0, 1, 2, …, n-1. He plans to spend m consecutive days（2 ≤ m ≤ n）in Beijing. During these m days, he intends to use the first day and another day to visit Peking university. Before he made his plan, Ming investigated on the number of tourists who would be waiting in line to enter Peking university during his n-day trip, and the results could be represented by an integer sequence p[i] (0 ≤ i ≤ n-1, p[i] represents the number of waiting tourists on day i). To save time, he hopes to choose two certain dates a and b to visit PKU(0 ≤ a < b ≤ n-1), which makes p[a] + p[b] as small as possible.

Unfortunately, Ming comes to know that traffic control will be taking place in Beijing on some days during his n-day trip, and he won’t be able to visit any place in Beijing, including PKU, on a traffic control day. Ming loves Beijing and he wants to make sure that m days can be used to visit interesting places in Beijing. So Ming made a decision:  spending k (m ≤ k ≤ n) consecutive days in Beijing is also acceptable if there are k - m traffic control days among those k days. Under this complicated situation, he doesn’t know how to make the best schedule. Please write a program to help Ming determine the best dates of the two days to visit Peking University.  Data guarantees a unique solution.

输入

There are no more than 20 test cases.

For each test case:

The first line contains two integers, above mentioned n and m (2 ≤ n ≤ 100, 2 ≤ m ≤ n).

The second line contains n integers, above mentioned p[0] , p[1] , … p[n-1]. (0 ≤ p[i] ≤ 1000, i = 0 ... n-1)

The third line is an integer q (0 ≤ q ≤ n), representing the total number of traffic control days during the n-day trip, followed by q integers representing the dates of these days.

输出

One line, including two integers a and b, representing the best dates for visiting PKU.

样例输入

7 3
6 9 10 1 0 8 35
3 5 6 2
4 2
10 11 1 2
1 2

样例输出

0 3
1 3

思路：

暴力枚举。

代码：

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

using namespace std;
int a[101];
int b[101];
int dp[101][101];
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    memset(dp,0,sizeof(dp));
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    int q;
    cin>>q;
    for(int i=1;i<=q;i++)
    {
        int w;
        scanf("%d",&w);
        b[w]=1;
    }

    int su=20000;
    int st=0,en=0;
    for(int i=0;i<n;i++)
    {   int k=1;
        int sum=20000;
        int p=0;
        int o=0;
        int g=0;
        if(b[i])continue;
        for(int j=i+1;j<n;j++)
        {
            if(b[j])continue;
            //int h=k;
            dp[i][j]=a[i]+a[j];
            /*/for(int l=i;l<n;l++)
            {    h++;
                if(b[l])continue;
                 if(l==j)continue;
                 dp[j][l]=a[j]+a[l];

                 if(h>=m){break;}
            }/*/
            if(sum>dp[i][j]){sum=dp[i][j];o=i;p=j;}

            k++;
            if(k>=m){g=1;break;}
        }

        if(su>sum&&g==1){st=o;en=p;su=sum;}
    }
    printf("%d %d\n",st,en);
    }
    return 0;
}