CF1677D Tokitsukaze and Permutations
题面传送门
继NOI online后冒泡排序的更新用法(雾
在NOI online那道题里面我们关注的是每个数字对应的逆序对数大小,现在我们要关注每个位置。
感觉上应该和上一个是相近的,我们考虑每个逆序对数为\(0\)的位置,它会一直往后直到碰到另一个逆序对数为\(0\)的位置,那么它会让其它非零的位置代表的逆序对数前移一位并减一了。
再来考虑\(0\),不难发现\(0\)的位置也是满足这个性质的,只不过第一个转到最后一个上去了。
然后每个位置对应的逆序对数的范围就可以限定,然后看成插入前面的就好了。
时间复杂度\(O(n)\)
code:
#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define ll long long
#define db double
#define lb long db
#define N (1000000+5)
#define M (220+5)
#define K (12+5)
#define mod 998244353
#define Mod (mod-1)
#define eps (1e-9)
#define ull unsigned ll
#define it iterator
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
#define Mc(x,y) memcpy(x,y,sizeof(x))
#define d(x,y) (n*(x-1)+(y))
#define R(n) (rand()*rand()%(n)+1)
#define Pc(x) putchar(x)
#define LB lower_bound
#define UB upper_bound
#define PB push_back
using namespace std;
int n,m,k,T,A[N],B[N];ll Ans;
I void Solve(){
int i,j;scanf("%d%d",&n,&k);for(i=1;i<=n;i++) scanf("%d",&A[i]);
Ans=1;for(i=1;i<=k;i++) B[i]=A[n-k+i];for(i=1;i<=n-k;i++) B[i+k]=A[i];
for(i=1;i<=n;i++){if(B[i]>0&&B[i]+k>=i) {puts("0");return;}
if(!B[i]) Ans=Ans*min(i,k+1)%mod;else if(B[i]==-1) Ans=Ans*i%mod;}printf("%lld\n",Ans);
}
int main(){
// freopen("1.in","r",stdin);
scanf("%d",&T);while(T--) Solve();
}