https://ac.nowcoder.com/acm/contest/57362/H
题意:
每次操作可以将1~k个数从小到大插入已知序列的任意位置。现在给你一个序列,问有多少种子序列可以通过任意次数操作获得。
做法:
显然每做一次操作1的数量必定增加,可以将1作为每一个操作块的开头,并用一个栈来储存1出现的次数,也就是操作次数(栈的大小是合法块的数量)。这样就可以在每一块里单独讨论合法的子序列之后再将其并到前面合法的序列后。我们用vector储存数k出现在第几个块,在往后计算到k+1时,取出v[k]最后一个位置,将它从v[k]中删除(相当于k无效),放入v[k+1]中。
这样将会出现三种情况:
1.本次操作1后的数都可以在自己的块内得到满足,只需要加上cnt(cnt为前面合法的能与当前块链接的块数,相当于合并)。因为1可以单独做一个序列,所以每读入一个1就加1次。
2.本次操作1后的数需要从前面的块获取,比如121233,第二个3就需要连接到第1个块后,此时第一个块是合法的,能和当前块连接,那就在栈中删除操作1后的块,此时块1 与块2 合并。(块1的合法依赖于块2)
3.本次操作1后的数需要从前面的块获取,但前面块不合法,比如1241233,在模拟到4的时候块1就被清除出栈了,所以无法在栈中找到1,此时cnt+=0;
代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define sz(x) (int)x.size()
const int N=1e6+5;
const int INF=1e18+7;
const int inf=1e9+7;
const int mod=998244353;
vector<int>v[N];
int s[N];
signed main()
{
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
int n;
cin>>n;
int tot=0,ans=0;
int top=0;
for(int i=1;i<=n;i++){
int x;
cin>>x;
if(x==1)v[1].push_back(++tot),s[++top]=tot,ans+=top;
else{
int now=0;
if(!v[x-1].empty()){//
now=v[x-1][v[x-1].size()-1];//获得有效x-1最后出现的位置
v[x-1].pop_back();
v[x].push_back(now);
}
while(s[top]>now)--top;//说明比x小的数在它前面的集合里
ans+=top;
//cout<<top<<endl;
}
}
cout<<ans<<endl;
}
思考:
比赛的时候想到了要用1分块和每次向前寻找后合并,但根本没有想到用栈储存块数和用vector存k-1,就写了非常复杂的代码。:(