1. 程式人生 > >luogu2564 [SCOI2009]生日禮物

luogu2564 [SCOI2009]生日禮物

markdown 生日禮物 枚舉 pac getc for har tchar 比較

排序枚舉左端點,則右端點必定不降

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
struct Node{
    int pos, val;
}nd[1000005];
int n, k, cnt[63], uu, vv=0, rig=0, re=0, ans=0x3f3f3f3f;
void rn(int &x){
    x = 0;
    char ch=getchar();
    while(ch<‘0‘
|| ch>‘9‘) ch = getchar(); while(ch>=‘0‘ && ch<=‘9‘){ x = x * 10 + ch - ‘0‘; ch = getchar(); } } bool cmp(Node x, Node y){ return x.pos<y.pos; } int main(){ rn(n); rn(k); for(int i=1; i<=k; i++){ rn(uu); for(int j=1; j<=uu; j++){ rn(nd[++vv].pos); nd[vv].val = i; } } sort(nd+1
, nd+1+vv, cmp); for(int i=1; i<=n; i++){ while(re<k && rig<n){ rig++; if(++cnt[nd[rig].val]==1) re++; } if(re==k) ans = min(ans, nd[rig].pos-nd[i].pos); if(--cnt[nd[i].val]==0) re--; } cout<<ans<<endl; return
0; }

當然也可以用滑動窗口的思想做,就是比較慢

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
struct Node{
    int pos, val;
}nd[1000005];
int n, k, cnt[63], uu, vv=0, lev=1, rig=0, a[1000005];
void rn(int &x){
    x = 0;
    char ch=getchar();
    while(ch<‘0‘ || ch>‘9‘) ch = getchar();
    while(ch>=‘0‘ && ch<=‘9‘){
        x = x * 10 + ch - ‘0‘;
        ch = getchar();
    }
}
bool cmp(Node x, Node y){
    return x.pos<y.pos;
}
bool check(int lim){
    memset(cnt, 0, sizeof(cnt));
    lev = 1, rig = 0;
    int re=0;
    for(int i=1; i<=n; i++){
        while(lev<=rig && nd[a[lev]].pos<nd[i].pos-lim){
            if(--cnt[nd[a[lev]].val]==0)    re--;
            lev++;
        }
        a[++rig] = i;
        if(++cnt[nd[i].val]==1) re++;
        if(re==k)   return true;
    }
    return false;
}
int main(){
    rn(n); rn(k);
    for(int i=1; i<=k; i++){
        rn(uu);
        for(int j=1; j<=uu; j++){
            rn(nd[++vv].pos);
            nd[vv].val = i;
        }
    }
    sort(nd+1, nd+1+vv, cmp);
    int l=0, r=nd[vv].pos, mid, ans=0;
    while(l<=r){
        mid = (l + r) >> 1;
        if(check(mid)){
            ans = mid;
            r = mid - 1;
        }
        else    l = mid + 1;
    }
    cout<<ans<<endl;
    return 0;
}

luogu2564 [SCOI2009]生日禮物