CF1540E Tasty Dishes [线性代数]

噫,好,线代题!

果然学了线代也还是不会做 /kk

思路

容易看出最优策略是什么。设 \(d_i\) 表示第 \(i\) 个人在哪天开始活过来。

因为一个人只能从负变正一次,所以 \(d\) 只会变化 \(O(n)\) 次。每次变化都可以 \(O(n^3)\) 重新得到 \(d\) 。所以我们不妨先假装 \(d\) 不变。

这时候就发现问题很大:他可以修改一次就询问一次,相当于可以任意询问第 \(k\) 个时刻,第 \(i\) 个人对第 \(j\) 个人的贡献。但是这样的状态数有 \(O(kn^2)\) 个,转移还需要 \(O(n)\) 时间,所以不太可能预处理。

转移写成矩阵?矩阵这么大一个,看起来也希望渺茫。

但是因为不用矩阵的做法陷入了死胡同,所以只能考虑分析转移矩阵的性质。

发现因为图是 DAG ,所以转移矩阵 \(A\) 是上三角矩阵。并且题设告诉我们对角线上恰好是 \(1\sim n\) ,

那么线代的知识告诉我们

  • 这个矩阵可以在 \(O(n^2)\) 的时间内高斯消元。
  • 这个矩阵会存在 \(n\) 个特征值,恰好是 \(1\sim n\) 。
  • 每个特征值的特征向量可以直接高斯消元求出来。

那么假设 \(\lambda_i=i\) 对应的某个特征向量是 \(v_i\) 。设 \(e_i=(0,0,\cdots,0,1,\cdots,0)\) 。

因为每个人的 \(d\) 不同,所以不太可能合在一起做,只能考虑每个人对答案的贡献。考虑第 \(i\) 个人,不妨假设 \(d_i\le k\) ,那么贡献显然是

\[e_{[l,r]}' A^{k-d_i}e_ia_i \]

因为有特征向量,所以可以先把每个 \(e_i\) 拆成 \(v\) 的线性表示。显然这只需要把 \(v\) 排在一起然后求逆即可。设 \(e_i=\sum_j c_{i,j}v_j\) ,那么上面的式子可以重新写成

\[a_i\sum_j c_{i,j}j^{k-d_i}(e_{[l,r]}'v_j) \]

此时可以把 \(i,j\) 的枚举顺序调换,变成枚举每个 \(j\) 的贡献。注意一下 \(d_i,k\) 的大小关系,就可以 \(O(\log n)\) 的时间得到每个 \(j\) 的贡献了。

\(d\) 的变化对上面这个过程的影响不是很大。

总复杂度 \(O(n^3+qn\log n)\) 。

代码

#include<bits/stdc++.h>
namespace my_std{
	using namespace std;
	#define pii pair<int,int>
	#define fir first
	#define sec second
	#define MP make_pair
	#define rep(i,x,y) for (int i=(x);i<=(y);i++)
	#define drep(i,x,y) for (int i=(x);i>=(y);i--)
	#define go(x) for (int i=head[x];i;i=edge[i].nxt)
	#define templ template<typename T>
	#define sz 333
	#define mod 1000000007ll
	typedef long long ll;
	typedef double db;
	mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
	templ inline T rnd(T l,T r) {return uniform_int_distribution<T>(l,r)(rng);}
	templ inline bool chkmax(T &x,T y){return x<y?x=y,1:0;}
	templ inline bool chkmin(T &x,T y){return x>y?x=y,1:0;}
	templ inline void read(T& t)
	{
		t=0;char f=0,ch=getchar();double d=0.1;
		while(ch>'9'||ch<'0') f|=(ch=='-'),ch=getchar();
		while(ch<='9'&&ch>='0') t=t*10+ch-48,ch=getchar();
		if(ch=='.'){ch=getchar();while(ch<='9'&&ch>='0') t+=d*(ch^48),d*=0.1,ch=getchar();}
		t=(f?-t:t);
	}
	template<typename T,typename... Args>inline void read(T& t,Args&... args){read(t); read(args...);}
	char __sr[1<<21],__z[20];int __C=-1,__zz=0;
	inline void Ot(){fwrite(__sr,1,__C+1,stdout),__C=-1;}
	inline void print(int x)
	{
		if(__C>1<<20)Ot();if(x<0)__sr[++__C]='-',x=-x;
		while(__z[++__zz]=x%10+48,x/=10);
		while(__sr[++__C]=__z[__zz],--__zz);__sr[++__C]='\n';
	}
	void file()
	{
		#ifdef NTFOrz
		freopen("a.in","r",stdin);
		#endif
	}
	inline void chktime()
	{
		#ifdef NTFOrz
		cerr<<clock()/1000.0<<'\n';
		#endif
	}
	#ifdef mod
	ll ksm(ll x,int y){ll ret=1;for (;y;y>>=1,x=x*x%mod) if (y&1) ret=ret*x%mod;return ret;}
	ll inv(ll x){return ksm(x,mod-2);}
	#else
	ll ksm(ll x,int y){ll ret=1;for (;y;y>>=1,x=x*x) if (y&1) ret=ret*x;return ret;}
	#endif
//	inline ll mul(ll a,ll b){ll d=(ll)(a*(double)b/mod+0.5);ll ret=a*b-d*mod;if (ret<0) ret+=mod;return ret;}
}
using namespace my_std;

int n;
ll a[sz]; int d[sz];
vector<int>E[sz];

ll A[sz][sz],V[sz][sz],C[sz][sz];
void init()
{
	rep(i,1,n) A[i][i]=i;
	rep(i,1,n) for (auto v:E[i]) A[i][v]=v;
	rep(i,1,n)
	{
		V[i][i]=1;
		drep(j,i,1) { if (j!=i) V[i][j]=mod-V[i][j]*inv(i-j)%mod; drep(k,j-1,1) (V[i][k]+=mod-V[i][j]*A[k][j]%mod)%=mod; }
	}
	rep(i,1,n)
	{
		C[i][i]=1;
		drep(j,i,1) drep(k,j-1,1) (C[i][k]+=mod-C[i][j]*V[j][k]%mod)%=mod;
	}
	rep(i,1,n) rep(j,1,n) (V[i][j]+=V[i][j-1])%=mod;
}
int id[sz],pos[sz],D[sz]; ll val[sz][sz];
ll tr[sz][sz];
void add(ll *tr,int x,ll w){while (x<=n) (tr[x]+=w)%=mod,x+=x&-x;}
ll query(ll *tr,int x){ll res=0;while (x) res+=tr[x],x-=x&-x;return res%mod;}
void rebuild()
{
	drep(i,n,1) if (a[i]>0) d[i]=0; else { d[i]=1e9; for (auto v:E[i]) chkmin(d[i],d[v]+1); }
	rep(i,1,n) id[i]=i; sort(id+1,id+n+1,[](int x,int y){return d[x]<d[y];}); rep(i,1,n) D[i]=d[id[i]],pos[id[i]]=i;
	rep(j,1,n) rep(i,1,n) val[j][i]=C[i][j]*ksm(j,mod-1-d[i])%mod,tr[j][i]=0;
	rep(j,1,n) rep(i,1,n) add(tr[j],pos[i],(a[i]+mod)*val[j][i]%mod);
}

int main()
{
	file();
	read(n);
	rep(i,1,n) read(a[i]);
	rep(i,1,n) { int c,x; read(c); while (c--) read(x),E[i].push_back(x); }
	init(); rebuild();
	int Q; read(Q);
	while (Q--)
	{
		int tp,x,y,z; read(tp,x,y);
		if (tp==1)
		{
			read(z);
			int p=lower_bound(D+1,D+n+1,x)-D-1; ll ans=0;
			rep(j,1,n) (ans+=(V[j][z]-V[j][y-1]+mod)*ksm(j,x)%mod*query(tr[j],p)%mod)%=mod;
			rep(i,y,z) if (d[i]>=x) (ans+=a[i]+mod)%=mod;
			printf("%lld\n",ans);
		}
		else { ll t=a[x]; a[x]+=y; if (t<=0&&a[x]>0) rebuild(); else rep(j,1,n) add(tr[j],pos[x],val[j][x]*y%mod); }
	}
	return 0;
}
上一篇:D. Shichikuji and Power Grid(Codeforces Round #597 (Div. 2)题解)


下一篇:[CF1491F]Magnets