2024牛客寒假算法基础集训营2--Tokitsukaze and Eliminate

题目

 输入数据

6
3
1 2 1
5
1 2 2 1 2
11
2 2 1 2 2 1 1 2 2 1 2
1
1
5
1 2 3 4 5
10
4 1 5 5 2 2 5 2 4 2

 输出数据

2
2
6
1
1
2

样例说明

第一组测试数据：
初始宝石为 [1,2,1][1,2,1][1,2,1]；
第 111 次操作选择颜色 222，可以消除最右边的 222 个宝石，当前剩余宝石为 [1][1][1]；
第 222 次操作选择颜色 111，即可把所有宝石都消除。
第三组测试数据：
初始宝石为 [2,2,1,2,2,1,1,2,2,1,2][2,2,1,2,2,1,1,2,2,1,2][2,2,1,2,2,1,1,2,2,1,2]；
第 111 次操作选择颜色 111，可以消除最右边的 222 个宝石，当前剩余宝石为 [2,2,1,2,2,1,1,2,2][2,2,1,2,2,1,1,2,2][2,2,1,2,2,1,1,2,2]；
第 222 次操作选择颜色 111，可以消除最右边的 333 个宝石，当前剩余宝石为 [2,2,1,2,2,1][2,2,1,2,2,1][2,2,1,2,2,1]；
第 333 次操作选择颜色 222，可以消除最右边的 222 个宝石，当前剩余宝石为 [2,2,1,2][2,2,1,2][2,2,1,2]；
第 444 次操作选择颜色 111，可以消除最右边的 222 个宝石，当前剩余宝石为 [2,2][2,2][2,2]；
第 555 次操作选择颜色 222，可以消除最右边的 111 个宝石，当前剩余宝石为 [2][2][2]；
第 666 次操作选择颜色 222，即可把所有宝石都消除。
第六组测试数据：
初始宝石为 [4,1,5,5,2,2,5,2,4,2][4,1,5,5,2,2,5,2,4,2][4,1,5,5,2,2,5,2,4,2]；
第 111 次操作选择颜色 555，可以消除最右边的 444 个宝石，当前剩余宝石为 [4,1,5,5,2,2][4,1,5,5,2,2][4,1,5,5,2,2]；
第 222 次操作选择颜色 444，即可把所有宝石都消除；
或者也可以这么操作：第 111 次操作选择颜色 111，第 222 次操作选择颜色 444；
还可以这么操作：第 111 次操作选择颜色 444，第 222 次操作选择颜色 444。

题解

代码一

#include<bits/stdc++.h>
#define int long long
#define pii pair<int,int>
using namespace std;


void solve(){
	int n; cin>>n;
    vector<int> re(n+1);
    map<int,int> mp;
    for(int i=1;i<=n;i++) cin>>re[i],mp[re[i]]++;
    int cnt=mp.size(),ans=0;
    set<int> s;
    for(int i=n;i>=1;i--){
        s.insert(re[i]);
        mp[re[i]]--;
        if(mp[re[i]]==0) mp.erase(re[i]);
        if(s.size()==cnt){
            ans++;
            cnt=mp.size();
            s.clear();
        }
    }
    cout<<ans<<"\n";
}

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	
	int t=1;
 	cin>>t;
	while(t--) solve();
}

代码二

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M=2e6+50;
int pre[M],a[M],las[M];
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            pre[i]=0;
            las[i]=0;
        }
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            pre[i]=las[a[i]];
            las[a[i]]=i;
        }
        int last=1e9,newlast=1e9;
        for(int i=1;i<=n;i++){
            if(las[i]!=0)last=min(last,las[i]);
        }
        int ans=0;
        for(int i=n;i>=1;i--){
//            printf("i=%d last=%d\n",i,last);
            if(i>last){
                if(pre[i]!=0)newlast=min(newlast,pre[i]);
            }else{
                if(pre[i]!=0)newlast=min(newlast,pre[i]);
                last=newlast;
                ans++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

代码三

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
int n,a[200005],vis[200005];
map<int,int>m;
void solve()
{
    cin>>n;
    int cnt=0,cnt1=0,cnt2=0,ans=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(!vis[a[i]])cnt++;
        vis[a[i]]++;
    }
    for(int i=n;i>=1;i--){
        m[a[i]]++;vis[a[i]]--;
        if(m[a[i]]==1)cnt1++;
        if(vis[a[i]]==0)cnt2++;
        if(cnt1==cnt){m.clear();ans++;cnt-=cnt2;cnt1=0;cnt2=0;}
    }
    cout<<ans<<endl;
    m.clear();
}
int main()
{
    int t;
    cin>>t;
    while(t--)
        solve();
    return 0;
}