【2020省选模拟】题解

T1：

首先判$s<0$或$s=0,a\ne 0$无解
考虑操作相当于交换前缀和
对于序列就是前缀和单调不降且非负
如果非负则答案为逆序对数

对于环，令前缀和为循环无限前缀和（包括负下标）
设$p_i={\sum }_{j>i,s_j<s_i}1$，则显然$p$有限
一次操作交换$a<0$的地方会恰好使$p$减一

所以只用计算$p$
对于$i<j<i+n,s_j<s_i$的$j$,贡献有$\frac{s_i-s_j-1}{s_n}+1$
可以看成若$s_i\%s_n>s_j\%s_n,$贡献为$\frac{s_i}{s_n}-\frac{s_j}{s_n}$
否则还要减一

考虑先算不考虑$i<j<i+n$的限制，那么最后就是对于$i<j,s_i>s_j$
会在$i$而非$i+n$计算，所以还要加上$1$

分别$O(nlogn)$做即可

#include<bits/stdc++.h>
using namespace std;
#define cs const
#define pb push_back
#define ll long long 
#define pii pair<int,int>
#define fi first
#define se second
#define bg begin
cs int RLEN=(1<<20)|3;
inline char gc(){
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob)?EOF:*ib++;
}
inline int read(){
	char ch=gc();
	int res=0;bool f=1;
	while(!isdigit(ch))f^=ch=='-',ch=gc();
	while(isdigit(ch))res=(res*10)+(ch^48),ch=gc();
	return f?res:-res;
}
inline ll readll(){
	char ch=gc();
	ll res=0;bool f=1;
	while(!isdigit(ch))f^=ch=='-',ch=gc();
	while(isdigit(ch))res=(res*10)+(ch^48),ch=gc();
	return f?res:-res;	
}
inline int readstring(char *s){
	char ch=gc();int l=0;
	while(isspace(ch))ch=gc();
	while(!isspace(ch)&&ch!=EOF)s[++l]=ch,ch=gc();
	return l;
}
inline char readchar(){
	char ch=gc();
	while(isspace(ch))ch=gc();return ch;
}
template<typename tp>inline void chemx(tp &a,tp b){a<b?a=b:0;}
template<typename tp>inline void chemn(tp &a,tp b){a<b?a=b:0;}
cs int N=200005;
int a[N],n,cnt;
struct bit{
	ll tr[N];
	#define lb(x) (x&(-x))
	void clear(){memset(tr,0,sizeof(tr));}
	void inss(int p,int k){for(;p>0;p-=lb(p))tr[p]+=k;}
	ll asks(int p,ll res=0){for(;p<=cnt;p+=lb(p))res+=tr[p];return res;}
	void insp(int p,int k){for(;p<=cnt;p+=lb(p))tr[p]+=k;}
	ll askp(int p,ll res=0){for(;p>0;p-=lb(p))res+=tr[p];return res;}
	#undef lb
}t1,t2;
int b[N],p[N];
struct opt{
	int s,a,b,vl;
	friend inline bool operator <(cs opt &a,cs opt &b){
		return a.s<b.s;
	}
}q[N];
inline void solve(){
	n=read();int s=0;
	for(int i=1;i<=n;i++)s+=read(),a[i]=s;
	if(s<0){puts("-1");return;}
	if(s==0){
		for(int i=1;i<=n;i++)if(a[i]){puts("-1");return;}
		puts("0");return;
	}cnt=0;ll res=0;
	for(int i=1;i<=n;i++)b[++cnt]=a[i],b[++cnt]=a[i]+s;
	sort(b+1,b+cnt+1),cnt=unique(b+1,b+cnt+1)-b-1;
	for(int i=1;i<=n;i++){
		q[i].s=(a[i]%s+s)%s+1,q[i].vl=(a[i]<0)?((a[i]-s+1)/s):(a[i]/s);
		q[i].a=lower_bound(b+1,b+cnt+1,a[i])-b,q[i].b=p[i]=lower_bound(b+1,b+cnt+1,a[i]+s)-b;
	}
	sort(q+1,q+n+1);
	for(int i=1,j;i<=n;i=j){
		j=i+1;
		while(j<=n&&q[i].s==q[j].s)j++;
		for(int x=i;x<j;x++)
			t1.inss(q[x].a,1),t2.inss(q[x].a,q[x].vl);
		for(int x=i;x<j;x++)
			res+=t2.asks(q[x].b+1)-t1.asks(q[x].b+1)*(q[x].vl+1);
	}
	t1.clear(),t2.clear();
	for(int i=n,j;i>=1;i=j){
		j=i-1;
		while(j&&q[i].s==q[j].s)j--;
		for(int x=i;x>j;x--)
			res+=t2.asks(q[x].b+1)-t1.asks(q[x].b+1)*(q[x].vl);
		for(int x=i;x>j;x--)
			t1.inss(q[x].a,1),t2.inss(q[x].a,q[x].vl);
	}
	t1.clear(),t2.clear();
	for(int i=n;i;i--){
		res+=t1.askp(p[i]-1);
		t1.insp(p[i],1);
	}t1.clear();
	cout<<res<<'\n';
}
int main(){
	#ifdef Stargazer
	freopen("lx.in","r",stdin);
	#endif
	int T=read();
	while(T--)solve();
	return 0;
}

T2：

状压$DP$，矩乘优化
先带$\omega$点值之后$IDFT$原多项式
似乎直接多项式矩乘还挺优秀的
又被$ldx$教育/kk

#include<bits/stdc++.h>
using namespace std;
#define cs const
#define pb push_back
#define ll long long 
#define pii pair<int,int>
#define fi first
#define se second
#define bg begin
cs int RLEN=(1<<20)|3;
inline char gc(){
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob)?EOF:*ib++;
}
inline int read(){
	char ch=gc();
	int res=0;bool f=1;
	while(!isdigit(ch))f^=ch=='-',ch=gc();
	while(isdigit(ch))res=(res*10)+(ch^48),ch=gc();
	return f?res:-res;
}
inline ll readll(){
	char ch=gc();
	ll res=0;bool f=1;
	while(!isdigit(ch))f^=ch=='-',ch=gc();
	while(isdigit(ch))res=(res*10)+(ch^48),ch=gc();
	return f?res:-res;	
}
template<typename tp>inline void chemx(tp &a,tp b){a<b?a=b:0;}
template<typename tp>inline void chemn(tp &a,tp b){a<b?a=b:0;}
cs int mod=998244353;
inline int add(int a,int b){return (a+b>=mod)?(a+b-mod):(a+b);}
inline int dec(int a,int b){return (a<b)?(a-b+mod):(a-b);}
inline int mul(int a,int b){static ll r;r=(ll)a*b;return (r>=mod)?(r%mod):r;}
inline void Add(int &a,int b){a=(a+b>=mod)?(a+b-mod):(a+b);}
inline void Dec(int &a,int b){a=(a<b)?(a-b+mod):(a-b);}
inline void Mul(int &a,int b){static ll r;r=(ll)a*b;a=(r>=mod)?(r%mod):r;}
inline int ksm(int a,int b,int res=1){for(;b;b>>=1,Mul(a,a))(b&1)&&(Mul(res,a),1);return res;}
inline int Inv(int x){return ksm(x,mod-2);}
int n,m,k,lim;
cs int M=(1<<17)|3;
int rev[M],f[M];
typedef unsigned long long uLL;
inline void init_rev(int lim){
	for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)*(lim>>1));
}
inline void IDFT(int *f,int lim){
	for(int i=0;i<lim;i++)if(i>rev[i])swap(f[i],f[rev[i]]);
	for(int w,a0,a1,wn,mid=1;mid<lim;mid<<=1){
		wn=ksm(3,(mod-1)/(mid<<1));
		for(int i=0;i<lim;i+=mid<<1){
			w=1;
			for(int j=0;j<mid;j++,Mul(w,wn)){
				a0=f[i+j],a1=mul(w,f[i+j+mid]);
				f[i+j]=add(a0,a1),f[i+j+mid]=dec(a0,a1);
			}
		}
	}
	reverse(f+1,f+lim);
	for(int i=0,iv=Inv(lim);i<lim;i++)Mul(f[i],iv);
}
namespace solve1{

cs int T=2;
uLL tp[T][T];
struct mat{
	int a[T][T];
	void clear(){memset(a,0,sizeof(a));}
	friend inline mat operator *(cs mat &a,cs mat &b){
		mat c;
		tp[0][0]+=(ll)a.a[0][0]*b.a[0][0];
		tp[0][1]+=(ll)a.a[0][0]*b.a[0][1];
		tp[0][0]+=(ll)a.a[0][1]*b.a[1][0];
		tp[0][1]+=(ll)a.a[0][1]*b.a[1][1];
		tp[1][0]+=(ll)a.a[1][0]*b.a[0][0];
		tp[1][1]+=(ll)a.a[1][0]*b.a[0][1];
		tp[1][0]+=(ll)a.a[1][1]*b.a[1][0];
		tp[1][1]+=(ll)a.a[1][1]*b.a[1][1];
		for(int i=0;i<T;i++)
		for(int j=0;j<T;j++)c.a[i][j]=tp[i][j]%mod,tp[i][j]=0;
		return c;
	}
	friend inline mat operator ^(mat a,int b){
		mat res=a;b--;
		for(;b;b>>=1,a=a*a)if(b&1)res=res*a;
		return res;
	}
}bas;
int cnt[10],a[10];
inline void main(){
	int wn=ksm(3,(mod-1)/lim);
	for(int w=1,i=0;i<lim;i++,Mul(w,wn)){
		bas.a[0][0]=1,bas.a[0][1]=w,bas.a[1][0]=1,bas.a[1][1]=1;
		bas=bas^m;f[i]=add(bas.a[0][0],bas.a[0][1]);
	}
	init_rev(lim),IDFT(f,lim);
	cout<<f[k]<<'\n';return;
}

}
namespace solve2{

cs int T=4;
uLL tp[T][T];
struct mat{
	int a[T][T];
	void clear(){memset(a,0,sizeof(a));}
	friend inline mat operator *(cs mat &a,cs mat &b){
		mat c;
		for(int i=0;i<T;i++)
		for(int k=0;k<T;k++)
		for(int j=0;j<T;j++)
		tp[i][j]+=(ll)a.a[i][k]*b.a[k][j];
		for(int i=0;i<T;i++)
		for(int j=0;j<T;j++)c.a[i][j]=tp[i][j]%mod,tp[i][j]=0;
		return c;
	}
	inline void init(int x){
		a[0][0]=1,a[0][1]=x,a[0][2]=x,a[0][3]=x;
		a[1][0]=1,a[1][1]=1,a[1][2]=x,a[1][3]=1;
		a[2][0]=1,a[2][1]=x,a[2][2]=1,a[2][3]=1;
		a[3][0]=1,a[3][1]=1,a[3][2]=1,a[3][3]=1;
	}
	friend inline mat operator ^(mat a,int b){
		mat res=a;b--;
		for(;b;b>>=1,a=a*a)if(b&1)res=res*a;
		return res;
	}
}bas;
inline void main(){
	int wn=ksm(3,(mod-1)/lim);
	for(int w=1,i=0;i<lim;i++,Mul(w,wn)){
		bas.init(w);
		bas=bas^m;
		for(int j=0;j<T;j++)
		Add(f[i],bas.a[0][j]);
	}
	init_rev(lim),IDFT(f,lim);
	cout<<f[k]<<'\n';return;
}

}
namespace solve3{

cs int T=9;
uLL tp[T][T];
struct mat{
	int a[T][T];
	void clear(){memset(a,0,sizeof(a));}
	friend inline mat operator *(cs mat &a,cs mat &b){
		mat c;
		for(int i=0;i<T;i++)
		for(int k=0;k<T;k++)
		for(int j=0;j<T;j++)
		tp[i][j]+=(ll)a.a[i][k]*b.a[k][j];
		for(int i=0;i<T;i++)
		for(int j=0;j<T;j++)c.a[i][j]=tp[i][j]%mod,tp[i][j]=0;
		return c;
	}
	inline void init(int x){
		int x2=mul(x,x),ix=Inv(x);
		int b[9][9]=
		{
			1,x,x,x,x,x,0,x2,x,
			1,1,x,x,1,x,0,x,1,
			1,x,1,x,1,1,0,x2,1,
			1,x,x,1,x,1,0,x,1,
			1,1,1,x,1,1,0,x,1,
			1,x,1,1,1,1,0,x,1,
			1,1,x,1,1,1,1,0,1,
			1,1,x,1,1,1,0,1,ix,
			1,1,1,1,1,1,1,0,1
		};
		memcpy(a,b,sizeof(b));
	}
	friend inline mat operator ^(mat a,int b){
		mat res=a;b--;
		for(;b;b>>=1,a=a*a)if(b&1)res=res*a;
		return res;
	}
}bas;
int cnt[10],a[10];
inline void main(){
	int wn=ksm(3,(mod-1)/lim);
	for(int w=1,i=0;i<lim;i++,Mul(w,wn)){
		bas.init(w);
		bas=bas^m;
		for(int j=0;j<T;j++)
		Add(f[i],bas.a[0][j]);
	}
	init_rev(lim),IDFT(f,lim);
	cout<<f[k]<<'\n';return;
}

}
int main(){
	n=read(),m=read(),k=read();
	lim=1;while(lim<=(n*m+1)/2+1)lim<<=1;
	if(n==1)solve1::main();
	if(n==2)solve2::main();
	if(n==3)solve3::main();
	return 0;
}

T3:

here