Toyota Programming Contest 2024#1（AtCoder Beginner Contest 337）F. Usual Color Ball Problems（思维题 双指针）

题目

长为n(n<=2e5)的球序列，第i个球颜色ci，有m(m<=n)个空盒子，编号1-m

对于r∈[0,n-1]的每个r，打印以下问题的答案：

（一）把序列中最前面的球移动到序列末尾，执行r次

（二）顺序遍历n个球，对于当前队首的球x，记颜色为c[x]

（1）如果存在某一个盒子当前装了[1,k)个颜色为c[x]的球，就把球x放入这个盒子

（2）如果没有这样的盒子：

   ①如果有空盒子，就把球x放入编号最小的空盒子

   ②否则，扔掉这个球，不放进任何一个盒子（注意扔掉的操作不会影响下一个r的询问）

思路来源

官方题解

题解

题意还挺绕的

2400的双指针，难绷，heltion starsilk cuiaoxiang pyy纷纷折戟

主要是没想到双指针，想到双指针了的话可能会好做一些

主要是这个似乎只能双指针，二分似乎都不太行

首先执行r次，意味着，循环询问，对于每个开头的长为n的序列都询问一次，所以序列扩成二倍

然后考虑双指针，双指针维护第一个不在盒子内的球的位置，注意到它是单增的

并且，一旦确定了不在盒子内的球的位置p，

那么前面在盒子内的每种颜色球的个数就是确定的，

这些球占的盒子的个数也是确定的，

位置在p位置之后，但因为和p位置之前放入盒子里的球同色，而能被继续放入的球，

也可以O(1)确定，

即，如果某种颜色总共有all个球，

而当前占了x个盒子，则这种颜色的球总共放了min(x*k,all)个

一旦确定了不在盒子内的球的位置p，前面在盒子里的同色的球的个数也能确定，

记颜色w的球在盒子里的球为now[w]个，则占了now[w]/k（向上取整）个盒子

所以，双指针维护第一个不在盒子里的球的位置即可

代码

// #include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<vector>
// #include<map>
// #include<random>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=2e5+10,M=4e5+10;
/*
col:当前几种颜色
now[i]:i当前占了几个值，从而确定i占了几个盒子，从而确定i总共有多少个值在盒子里，每次加/删一个值最多改变一个盒子
1 1 1 2 3 
双指针维护连续的一段在盒子里的球 即第一个不在盒子里的球的位置 删球指针一定是不会向左挪的
*/
int n,m,k,cnt[N],c[M],col,sum,now[N];
void add(int w,int v){
    int x=(now[w]+k-1)/k;//w当前占了的盒子数
    col-=x;
    sum-=min(x*k,cnt[w]);//w当前在盒子里的数量
    now[w]+=v;
    x=(now[w]+k-1)/k;
    col+=x;
    sum+=min(x*k,cnt[w]);
}
int main(){
    sci(n),sci(m),sci(k);
    rep(i,1,n){
        sci(c[i]);
        c[n+i]=c[i];
        cnt[c[i]]++;
    }
    int r=0;
    rep(l,1,n){
        while(r+1<=l+n-1 && col<m){
            add(c[++r],1);
        }
        printf("%d\n",sum);
        add(c[l],-1);
    }
    return 0;
}