牛客练习赛 42 C.出题的诀窍(思维 + 实现 (卡常))

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
很容易想到按数值划分,计算每个数值的贡献次数。

如何计算一个数值的贡献次数:一个做法是枚举这个数值第一次在哪一行出现,计算有多少种下标可以使得这个数值第一次出现在这一行,将所有的情况加起来,正确性显然。实现的话可以从后往前扫一遍,借助hash表可以做到 O ( n 2 ) O(n^2) O(n2),使用unordered_map是过不去的。。。

另外一种计算贡献次数的方法:用总次数 - 当前数值不出现的次数。将每个数值存成一个pair,first代表数值,second代表所在行号,sort一下将整个序列按值分块,块内部按行号分块。外层循环扫值块,内存循环扫行号块,求出该数值不会出现的次数即可,虽然有排序但常数非常小。

还需要一个快读,否则超时


#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e3 + 10;
const int mod = 1e9 + 7;
#define pii pair<int,int>
#define fir first
#define sec second
int n,m;
vector<pii> h;
int pre[maxn];
void read(int &x) {
    x = 0; char c = getchar();
    while (c < '0' || c > '9') c = getchar();
    while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
}
int main() {
    read(n);read(m);
    for(int i = 1; i <= m; i++)
        for(int j = 1,x; j <= n; j++) {
            read(x);
            h.push_back(pii(x,i));
        }
    pre[0] = 1;
    for(int i = 1; i <= m; i++)
        pre[i] = 1ll * pre[i - 1] * n % mod;
    sort(h.begin(),h.end());
    int ans = 0;
    for(int l = 0,r; l < h.size(); l = r + 1) {
        r = l;
        while(r + 1 < h.size() && h[r + 1].fir == h[r].fir) r++;
        int cnt = 0,res = 1;
        for(int ll = l,rr; ll <= r; ll = rr + 1) {
            rr = ll;
            cnt++;
            while(rr + 1 <= r && h[rr + 1].sec == h[rr].sec) rr++;
            res = 1ll * res * (n - (rr - ll + 1)) % mod;
        }
        res = 1ll * res * pre[m - cnt] % mod;
        ans = (ans + 1ll * (pre[m] - res + mod) % mod * h[l].fir % mod) % mod;
    }
    printf("%d\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值