Codeforces 1042 E. Vasya and Magic Matrix —— 概率DP，Sorting

This way

题意：

给你一个矩阵，每个位置上都会有一个值，从一个点开始，每次会往值比他小的地方等概率地走，问你到最终走的距离（欧几里得距离）的长度期望是多少

题解：

很明显这是一张有向无环图，但是边数太多，所以不能建图。直接sort一下就知道有多少数比当前的数小了。从后往前做，到了当前点的时候，如果for一遍去转移的话那将是$O((nm{)}^2)$的时间复杂度，所以它必可以记录一个值进行转移
我们要求的东西是这个样子的：（假设起始位置经过排序之后是p）
$$E(p)=\sum _{i=1\&\&a_i<a_p}\frac{E(i)+(x_p-x_i{)}^2+(y_p-y_i{)}^2}{num[p]}$$
然后展开可得：
$$E(p)=\frac{1}{num[p]}(\sum _{i=1\&\&a_i<a_p}(E(i)+x_i^2+y_i^2-2x_i{x}_p-2y_i{y}_p))+x_p^2+y_p^2$$
那么我用s_2记录$x_i^2+y_i^2$，用sx,sy记录x和y的和，num记录当前位置可以转移的点的个数，sum表示E(i)的和，ans表示当前的答案。
然后我们可以知道，有可能有多个位置的值会相同，所以第i个位置能够转移到的地方并不是i-1，所以我在变量名前面加了"t_"表示这个是中介值

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pa pair<int,int>
const int N=1e3+5;
const ll mod=998244353;
struct node{
    ll v,x,y;
    bool operator< (const node& a)const {
        return v<a.v;
    }
}a[N*N];
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
ll dp[N*N];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    int all=0;
    int r,c;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            all++,scanf("%d",&a[all].v),a[all].x=i,a[all].y=j;
    sort(a+1,a+1+all);
    scanf("%d%d",&r,&c);
    ll s_2=0,sx=0,sy=0,num=0,sum=0,ans=0;
    ll t_s_2=0,t_sx=0,t_sy=0,t_num=0,t_sum=0;
    for(int i=1;i<=all;i++){
        if(a[i].v!=a[i-1].v){
            s_2=t_s_2;
            sx=t_sx;
            sy=t_sy;
            num=t_num;
            sum=t_sum;
        }
        ans=((sum+s_2-2*a[i].x*sx%mod-2*a[i].y*sy%mod+(a[i].x*a[i].x+a[i].y*a[i].y)*num%mod)%mod+mod)*qpow(num,mod-2)%mod;
        t_s_2=(t_s_2+a[i].x*a[i].x+a[i].y*a[i].y)%mod;
        t_sx=(t_sx+a[i].x)%mod;
        t_sy=(t_sy+a[i].y)%mod;
        t_sum=(t_sum+ans)%mod;
        t_num++;
        if(a[i].x==r&&a[i].y==c)break;
    }
    printf("%lld\n",ans);
    return 0;
}