洛谷-Vasya and Magic Matrix-(期望dp+欧几里得距离式子)

CF1042E

题意：
就是给你一个n✖m的矩阵，每个点有个值，给定一个出发点,每次可以等概率的移动到一个权值小于当前点权值的点,同时得分加上两个点之间欧几里得距离((x1-x2)×(x1-x2)+(y1-y2)×(y1-y2))的平方，直到走到不能能走为止，问你分数的期望。

思考：

1. 当时做的时候，求这个分数期望，那么我就想从起点开始往后走，走到这个点的概率是多少那么可以加多少分，但是我发现，这不好处理呀，因为这就是不同的路径不同的值。其实可以先按每个点的权值排序，然后每个点从前面所有大于他的点来转移，如果i点从前面所有的j来转移，那么不用的j，就代表从j跳到i。处理完每个点的dp值以后，把所有的最小的dp值都加上就可以了。
2. 当然从大的往小的走可以走到很多个终点，如果从小的往大的走，就会走到最大的那个，所以倒着推和正着推都一样。对于公式，推一推：
3. 值得注意的就是dp要开一个数组维护前缀和，不能把dp弄成前缀和，这样当处理权值相同的点的时候不好处理了。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 1e6+10,M = 2010;

struct Node{
	int a,b,c;
}va[N];

int T,n,m,k;
int vb[M][M];
int sumx[N],sumy[N],sumxx[N],sumyy[N],cnt;
int dp[N],sum[N];

bool cmp(Node A,Node B)
{
	return A.c<B.c;
}

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

void f(int i,int pre)
{
	int res = pre*(va[i].a*va[i].a%mod+va[i].b*va[i].b%mod)%mod;
	res = (res-2*va[i].a%mod*sumx[pre]%mod)%mod;
	res = (res-2*va[i].b%mod*sumy[pre]%mod)%mod;
	res = (res+(sumxx[pre]+sumyy[pre])%mod)%mod;
	dp[i] = (dp[i]+res+sum[pre])%mod*ksm(pre,mod-2)%mod;
	sum[i] = (sum[i-1]+dp[i])%mod;
}


signed main()
{
	IOS;
	cin>>n>>m;
	int stx,sty,value;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			cin>>vb[i][j];
			va[++cnt] = {i,j,vb[i][j]};
		}
	}
	cin>>stx>>sty;
	sort(va+1,va+1+cnt,cmp);
	for(int i=1;i<=cnt;i++)
	{
		sumx[i] = (sumx[i-1]+va[i].a)%mod;
		sumy[i] = (sumy[i-1]+va[i].b)%mod;
		sumxx[i] = (sumxx[i-1]+va[i].a*va[i].a%mod)%mod;
		sumyy[i] = (sumyy[i-1]+va[i].b*va[i].b%mod)%mod;
	}
	int pre = 0;
	for(int i=1;i<=cnt;i++)
	{
		if(va[i].c=va[i-1].c) pre = i-1;
		f(i,pre);
		if(va[i].a==stx&&va[i].b==sty)
		{
			cout<<(dp[i]%mod+mod)%mod;
			break;
		}
	}
	return 0;
}

总结：
多推推式子，转化题目的处理方式。对于处理一段区间的平方数什么什么的，都是有公式的一般。