2021年广东工业大学第十五届文远知行杯程序设计竞赛 G.分割（思维）

题意：

解法：

题目是要任取两条横线,两条竖线,将所有可能的方案组成的矩形面积相加.

容易发现横线的选择和竖线的选择是相互独立的,
因此我们只需要计算出选择两条横线能组成的所有矩形边长的和,
同时计算出选择两条竖线能组成的所有矩形变成的和,
两者相乘就是答案.

现在问题变为如何计算所有能组成的边长的和.

对于x[i],对答案的贡献为x[i]-x[1],x[i]-x[2],...x[i]-x[i-1],这些数的和.
合并一下就是x[i]*(i-1)-sum(x[1]+x[2]...x[i-1]),
维护一个前缀sum就可以O(n)计算了.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=1e6+5;
const int mod=1e9+7;
int x[maxm];
int y[maxm];
int n,m;
void solve(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>x[i];
    }
    for(int i=1;i<=m;i++){
        cin>>y[i];
    }
    sort(x+1,x+1+n);
    sort(y+1,y+1+m);
    int sum1=0;
    int ans1=0;
    for(int i=1;i<=n;i++){
        ans1+=x[i]*(i-1)%mod-sum1;
        ans1%=mod;
        sum1+=x[i];
        sum1%=mod;
    }
    int sum2=0;
    int ans2=0;
    for(int i=1;i<=m;i++){
        ans2+=y[i]*(i-1)%mod-sum2;
        ans2%=mod;
        sum2+=y[i];
        sum2%=mod;
    }
    int ans=ans1*ans2%mod;
    ans=(ans%mod+mod)%mod;
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---