uva 11536 Smallest Sub-Array

最新推荐文章于 2022-10-18 15:38:23 发布

原创最新推荐文章于 2022-10-18 15:38:23 发布 · 386 阅读

CC 4.0 BY-SA版权

本文讨论了一个关于序列生成与滑动窗口的问题，详细介绍了如何使用栈作为窗口存储载体，通过比较当前数与目标值K的关系，来确定最小连续序列长度，使得序列中包含1到K的所有值。代码实现包含初始化序列、使用队列和栈进行窗口操作，并在满足条件时更新答案。

题意:就是给定一个n个数字的序列，然后给定下面生成n个数字序列的m值，再给定k值，问最少的连续序列长度使得其中包含1……k所有值。

X1 = 1X2 = 2X3 = 3

Xi = (Xi−1 + Xi−2 + Xi−3)%M + 1 for i = 4 to N

此题应该是窗口滑动问题的经典了吧。

做法如下:使用栈来当窗口的存储载体。

对于当前的num[i] , 如果其小于K ，且队列的首部等于它，那么队首元素，然后检查能否在弹出对手元素，知道不能弹了（条件是：队首元素t小于等于k，且窗口中仅含有一个t）。然后就是看能不能跟新ans值了。（ans代表答案）。

如果大于k那么直接压栈。

注意两点：更新ans值时，必须得确定窗口中包含了1---k的所有值。

//
//  main.cpp
//  uva  11536 - Smallest Sub-Array
//
//  Created by XD on 15/8/17.
//  Copyright (c) 2015年 XD. All rights reserved.
//


#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include<vector>
#include <string.h>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <cstdio>
using namespace std ;
const int  maxn = 1000000 + 5 ;
int num[maxn] ;
int n ,m  , k ,s;
int min(int x , int y )
{
    return x > y ? y:x  ;
}
void init()
{
    num[1] = 1 ;num[2] = 2 ;
    num[3] = 3 ;
    for (int i = 4; i <= n; i++) {
        num[i] = (num[i-1] + num[i-2] + num[i-3]) % m + 1 ;
    }
}
int vis[1010] ;

int main(int argc, const char * argv[]) {
    int  T,ans ,tans;scanf("%d" ,&T) ;
    int kase =0 ;
    while (T--) {
        queue<int> q ;
        memset(vis, 0, sizeof(vis)) ;
        tans = ans = 0 ;
        scanf(" %d%d%d" ,&n ,&m,&k) ;
        printf("Case %d: " ,++kase) ;
//        if (k > m) {
//            printf("sequence nai\n") ;
//            continue ;
//        }
        init() ;
        while (!q.empty()) {
            q.pop() ;
        }
        int  t_k = 0 ;
        ans = maxn ;
        for (int i = 1;  i <=n ; i++) {
            if (num[i] <= k) {
                if (!q.empty() && num[i] == q.front()) {
                    vis[num[i]]++ ;
                    q.push(num[i]) ;
                    while (vis[q.front()]>1 || q.front() > k ) {
                        vis[q.front()]-= 1 ;
                        q.pop() ;
                    }
//                    tans =  ;
                    if(t_k == k ) ans = min((int)q.size() , ans) ;
                }
                else{
                    q.push(num[i]) ;
                    if (vis[num[i]] == 0) {
                        t_k++ ;
                    }
                    if (t_k==k) {
                        ans = min(ans , (int)q.size()) ;
                    }
                    vis[num[i]]+=1 ;
                }
            }
            else{
                q.push(num[i]) ;
            }
        }
        if (t_k==k) {
            printf("%d\n" ,ans ) ;
        }
        else{
            printf("sequence nai\n") ;
        }
    }
    return 0;
}