标签: 数据结构

[SDOI2014]数表 题解

[SDOI2014]数表 题解

题目地址:洛谷:【P3312】[SDOI2014]数表 – 洛谷、BZOJ:Problem 3529. — [Sdoi2014]数表

题目描述

有一张N*m的数表,其第i行第j列(1 < =i < =n,1 < =j < =m)的数值为能同时整除i和j的所有自然数之和。给定a,计算数表中不大于a的数之和。

输入输出格式

输入格式:
输入包含多组数据。
输入的第一行一个整数Q表示测试点内的数据组数
接下来Q行,每行三个整数n,m,a(|a| < =10^9)描述一组数据。

输出格式:
对每组数据,输出一行一个整数,表示答案模2^31的值。

输入输出样例

输入样例#1:

2
4 4 3
10 10 5

输出样例#1:

20
148

说明

1 < =N.m < =10^5 , 1 < =Q < =2*10^4

题解

参考资料:题解 P3312 【[SDOI2014]数表】 – ldlsn 的博客 – 洛谷博客
神奇的反演。
冷静分析一下,如果固定i和j,如何求能同时整除i和j的所有自然数之和。规定F(x)为x的因数和,例如F(2)=1+2=3, F(6)=1+2+3+6=12,则上面这个问题的答案就是F[\mathrm{gcd}(i, j)]
得到了上面的结论后,不考虑a的限制,我们可以知道答案就是\sum_{i=1}^n \sum_{j=1}^m F[\mathrm{gcd}(i, j)],下面就是对这个答案式的变形。
我们知道常用的反演结论g(x) = \sum_{i=1}^n \sum_{j=1}^m [\mathrm{gcd}(i, j) == x] = \sum_{i|d} \mu \left( \frac{d}{i} \right) \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor,直接套进答案里
\sum_{i=1}^n F(i)g(i) = \sum_{i=1}^n F(i) \sum_{i|d} \mu \left( \frac{d}{i} \right) \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor
改变一下求和顺序
\sum_{d=1}^n \left\lfloor \frac{n}{d} \right\rfloor \left\lfloor \frac{m}{d} \right\rfloor \sum_{i|d} F(i) \mu \left( \frac{d}{i} \right)
如果我们可以计算出f(x) = \sum_{i|x} F(i) \mu(\frac{x}{i})的值前缀和,那么可以除法分块做,而F(x)可以枚举倍数求和,这些都不是问题。
接下来我们来看带上限制应该怎么做,考虑离线处理,把询问按a限制从小到大排序,每一次将符合a条件的F(x)f(x)产生的贡献扔进一个维护前缀和的数据结构中(比如树状数组),为回答询问的除法分块求前缀和做准备。
至于那个奇怪的模数,自然溢出后&0x7fffffff就好。
总复杂度是O(n \sqrt{n} \log n)

代码

// Code by KSkun, 2018/6
#include <cstdio>
#include <cctype>

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) 
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 100005;

int prime[MAXN], mu[MAXN], ptot;
bool notprime[MAXN];

inline void sieve() {
    notprime[1] = true; mu[1] = 1;
    for(int i = 2; i < MAXN; i++) {
        if(!notprime[i]) {
            prime[++ptot] = i;
            mu[i] = -1;
        }
        for(int j = 1; j <= ptot && i * prime[j] < MAXN; j++) {
            int t = i * prime[j];
            notprime[t] = true;
            if(i % prime[j]) {
                mu[t] = -mu[i];
            } else {
                mu[t] = 0; break;
            }
        }
    }
}

int bit[MAXN];

inline int lowbit(int x) {
    return x & -x;
}

inline void add(int x, int v) {
    for(int i = x; i < MAXN; i += lowbit(i)) {
        bit[i] += v;
    }
}

inline int query(int x) {
    int res = 0;
    for(int i = x; i; i -= lowbit(i)) {
        res += bit[i];
    }
    return res;
}

struct Node {
    int x, v;
    inline bool operator<(const Node &rhs) const {
        return v < rhs.v;
    }
} F[MAXN];

inline void calf() {
    for(int i = 1; i <= 100000; i++) {
        F[i].x = i;
        for(int j = 1; i * j <= 100000; j++) {
            F[i * j].v += i;
        }
    }
    std::sort(F + 1, F + 100001);
}

inline int calc(int n, int m) {
    int ans = 0;
    if(n > m) std::swap(n, m);
    for(int i = 1; i <= n;) {
        int j = std::min(n / (n / i), m / (m / i));
        ans += (query(j) - query(i - 1)) * (n / i) * (m / i);
        i = j + 1;
    }
    return ans;
}

struct Query {
    int n, m, a, id;
} qur[MAXN];

inline bool qcmp(Query a, Query b) {
    return a.a < b.a;
}

int T, ans[MAXN];

int main() {
    sieve(); calf();
    T = readint();
    for(int i = 1; i <= T; i++) {
        qur[i].n = readint(); qur[i].m = readint(); qur[i].a = readint(); qur[i].id = i;
    }
    std::sort(qur + 1, qur + T + 1, qcmp);
    int now = 0;
    for(int i = 1; i <= T; i++) {
        while(now < 100000 && F[now + 1].v <= qur[i].a) {
            now++;
            for(int j = 1; j * F[now].x <= 100000; j++) {
                add(j * F[now].x, mu[j] * F[now].v);
            }
        }
        ans[qur[i].id] = calc(qur[i].n, qur[i].m);
    }
    for(int i = 1; i <= T; i++) {
        printf("%d\n", ans[i] & 0x7fffffff);
    }
    return 0;
}
[POI2005]SAM-Toy Cars 题解

[POI2005]SAM-Toy Cars 题解

题目地址:洛谷:【P3419】[POI2005]SAM-Toy Cars – 洛谷、BZOJ:Problem 1528. — [POI2005]sam-Toy Cars

题目描述

Jasio 是一个三岁的小男孩,他最喜欢玩玩具了,他有n 个不同的玩具,它们都被放在了很高的架子上所以Jasio 拿不到它们. 为了让他的房间有足够的空间,在任何时刻地板上都不会有超过k 个玩具. Jasio 在地板上玩玩具. Jasio’的妈妈则在房间里陪他的儿子. 当Jasio 想玩地板上的其他玩具时,他会自己去拿,如果他想玩的玩具在架子上,他的妈妈则会帮他去拿,当她拿玩具的时候,顺便也会将一个地板上的玩具放上架子使得地板上有足够的空间. 他的妈妈很清楚自己的孩子所以他能够预料到Jasio 想玩些什么玩具. 所以她想尽量的使自己去架子上拿玩具的次数尽量的少,应该怎么安排放玩具的顺序呢?

题意简述

Jasio有n个不同的玩具,有一段时间p内,每一单位时间他都想玩一个玩具,而地上只能放最多k个玩具。如果玩具不在地上,则他妈妈要帮他从架子上拿,如果此时地板上的玩具已经达到了k个,则还要拿一个玩具放回架子。求一种安排方式使得妈妈从架子上拿玩具的次数最少。

输入输出格式

输入格式:
第一行三个整数: n, k, p (1 <= k <= n <= 100.000, 1 <= p <= 500.000), 分别表示玩具的总数,地板上玩具的最多个数以及Jasio 他想玩玩具的序列的个数,接下来p行每行描述一个玩具编号表示Jasio 想玩的玩具.

输出格式:
一个数表示Jasio 的妈妈最少要拿多少次玩具.

输入输出样例

输入样例#1:

3 2 7
1
2
3
1
3
1
2

输出样例#1:

4

题解

贪心的结论是,放回去的一定是此时地上的“下一次使用的时间最晚的”玩具,这个信息可以在读入的时候用链表处理出来。因为当一个玩具还在地上的时候,从此时到下次使用它都不会再满足条件,反而还会占用一个位置,显然把这个位置让给现在用的玩具是最优的。
我们可以维护一个大根堆,按照每个玩具下一次被使用的时间来排序。但是注意,当一个玩具已经在地板上的时候,我们是无法更新这个玩具在堆中的下一次使用时间的信息的,因此我们得往堆中插入新的信息。此时可以用一个很取巧的办法,即对k加1,因为旧信息在以后永远都不会被移除,我们又无法改变堆的size值来修正有效信息的个数,就可以扩大k的值了。
复杂度O(n \log n)

代码

// Code by KSkun, 2018/6
#include <cstdio>
#include <cctype>

#include <algorithm>
#include <queue>
#include <vector>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) 
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 500005;

bool inque[MAXN];

int n, k, p, a[MAXN], nxt[MAXN], head[MAXN];

struct cmp {
    inline bool operator()(const int &a, const int &b) {
        return nxt[a] < nxt[b];
    }
};

std::priority_queue<int, std::vector<int>, cmp> pq;

int main() {
    n = readint(); k = readint(); p = readint();
    for(int i = 1; i <= p; i++) {
        a[i] = readint();
        nxt[head[a[i]]] = i; head[a[i]] = i;
    }
    for(int i = 1; i <= n; i++) {
        nxt[head[i]] = 1e9;
    }
    int ans = 0;
    for(int i = 1; i <= p; i++) {
        if(!inque[a[i]]) {
            if(pq.size() == k) {
                inque[a[pq.top()]] = false; pq.pop();
            }
            inque[a[i]] = true; ans++;
        } else {
            k++;
        }
        pq.push(i);
    }
    printf("%d", ans);
    return 0;
}
[TJOI2017]不勤劳的图书管理员 题解

[TJOI2017]不勤劳的图书管理员 题解

题目地址:洛谷:【P3759】[TJOI2017]不勤劳的图书管理员 – 洛谷、BZOJ:Problem 4889. — [Tjoi2017]不勤劳的图书管理员

题目描述

加里敦大学有个帝国图书馆,小豆是图书馆阅览室的一个书籍管理员。他的任务是把书排成有序的,所以无序的书让他产生厌烦,两本乱序的书会让小豆产生这两本书页数的和的厌烦度。现在有n本被打乱顺序的书,在接下来m天中每天都会因为读者的阅览导致书籍顺序改变位置。因为小豆被要求在接下来的m天中至少要整理一次图书。小豆想知道,如果他前i天不去整理,第i天他的厌烦度是多少,这样他好选择厌烦度最小的那天去整理。

题意简述

给一个数组,数组中一对逆序对的贡献是逆序对代表的数字的权值和,现在有m次操作,每次操作交换数组中两个位置的数字,要求输出每次交换完成后的上述贡献总和。

输入输出格式

输入格式:
第一行会有两个数,n,m分别表示有n本书,m天
接下来n行,每行两个数,ai和vi,分别表示第i本书本来应该放在ai的位置,这本书有vi页,保证不会有放置同一个位置的书
接下来m行,每行两个数,xj和yj,表示在第j天的第xj本书会和第yj本书会因为读者阅读交换位置

输出格式:
一共m行,每行一个数,第i行表示前i天不去整理,第i天小豆的厌烦度,因为这个数可能很大,所以将结果模10^9 +7后输出

输入输出样例

输入样例#1:

5 5
1 1
2 2
3 3
4 4
5 5
1 5
1 5
2 4
5 3
1 3

输出样例#1:

42
0
18
28
48

说明

对于20%的数据,1 ≤ ai; xj; yj ≤ n ≤ 5000, m ≤ 5000, vi ≤ 10^5
对于100%的数据,1 ≤ ai; xj; yj ≤ n ≤ 50000, m ≤ 50000, vi ≤ 10^5

题解

动态逆序对?我们想到了这个题[CQOI2011]动态逆序对。其实可以照搬这个题的树套树。
下面的内容默认你会用树套树解决上面那个题了。
我们考虑带权逆序对怎么解决。我们可以把权值也塞进线段树中,每次查询查找区间比某个数大/小的数字的个数以及权值和,个数*该数权值+权值和就是这个数与其他数字能产生的权值总和。
交换一对数字的时候,我们考虑先把这两个数字对答案的贡献减掉,然后从线段树里面抹去,再重新插入,加入贡献。注意如果先操作线段树再计算会造成重复计算,因此要操作一个加减一次,详细见代码。
总复杂度为O(n \log^2 n)

代码

// Code by KSkun, 2018/5
#include <cstdio>
#include <cctype>
#include <cstring>

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) 
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    register char c = fgc();
    while(!isdigit(c)) {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(isdigit(c)) {
        res = (res << 1) + (res << 3) + c - '0';
        c = fgc();
    }
    return res * neg;
}

const int MAXN = 50005, MO = 1e9 + 7;

struct Node {
    int lch, rch, cnt; LL sum;
    inline Node operator+(const Node &rhs) const {
        Node res = *this;
        res.cnt += rhs.cnt;
        res.sum += rhs.sum;
        return res;
    }
    inline Node& operator+=(const Node &rhs) {
        return *this = *this + rhs;
    }
    inline Node operator-(const Node &rhs) const {
        Node res = *this;
        res.cnt -= rhs.cnt;
        res.sum -= rhs.sum;
        return res;
    }
    inline Node& operator-=(const Node &rhs) {
        return *this = *this - rhs;
    }
} tr[MAXN * 200];
int rt[MAXN], tot;

int sta[MAXN], stop;

inline int newnode() {
    if(!stop) return ++tot;
    int p = sta[--stop];
    memset(tr + p, 0, sizeof(Node));
    return p;
}

inline void delnode(int p) {
    if(!p) return;
    sta[stop++] = p;
}

inline void insert(int &o, int l, int r, int x, LL v) {
    int p = newnode(); tr[p] = tr[o]; delnode(o); o = p;
    tr[o].cnt++; tr[o].sum += v;
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(x <= mid) insert(tr[o].lch, l, mid, x, v);
    else insert(tr[o].rch, mid + 1, r, x, v);
}

inline void erase(int &o, int l, int r, int x, LL v) {
    int p = newnode(); tr[p] = tr[o]; delnode(o); o = p;
    tr[o].cnt--; tr[o].sum -= v;
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(x <= mid) erase(tr[o].lch, l, mid, x, v);
    else erase(tr[o].rch, mid + 1, r, x, v);
}

inline Node querylar(int o, int l, int r, int x) {
    if(l == r) return Node {0, 0, 0, 0};
    int mid = (l + r) >> 1;
    if(x <= mid) return tr[tr[o].rch] + querylar(tr[o].lch, l, mid, x);
    else return querylar(tr[o].rch, mid + 1, r, x);
}

inline Node querysma(int o, int l, int r, int x) {
    if(l == r) return Node {0, 0, 0, 0};
    int mid = (l + r) >> 1;
    if(x <= mid) return querysma(tr[o].lch, l, mid, x);
    else return tr[tr[o].lch] + querysma(tr[o].rch, mid + 1, r, x);
}

int n, m;

inline int lowbit(int x) {
    return x & -x;
}

inline void add(int x, int a, LL p) {
    for(int i = x; i <= n; i += lowbit(i)) {
        insert(rt[i], 1, n, a, p);
    }
}

inline void erase(int x, int a, LL p) {
    for(int i = x; i <= n; i += lowbit(i)) {
        erase(rt[i], 1, n, a, p);
    }
}

inline LL query(int x, int a, LL p) {
    LL res = 0; Node tmp;
    for(int i = x; i; i -= lowbit(i)) {
        tmp = querylar(rt[i], 1, n, a);
        res += tmp.sum + tmp.cnt * p;
        res %= MO;
    }
    x += 0;
    for(int i = n; i; i -= lowbit(i)) {
        tmp = querysma(rt[i], 1, n, a);
        res += tmp.sum + tmp.cnt * p;
        res %= MO;
    }
    x += 0;
    for(int i = x; i; i -= lowbit(i)) {
        tmp = querysma(rt[i], 1, n, a);
        res -= tmp.sum + tmp.cnt * p;
        res = (res % MO + MO) % MO;
    }
    x += 0;
    return res;
}

int a[MAXN], v[MAXN], x, y;

int main() {
    n = readint(); m = readint();
    LL ans = 0;
    for(int i = 1; i <= n; i++) {
        a[i] = readint(); v[i] = readint();
        add(i, a[i], v[i]);
        ans += query(i, a[i], v[i]); ans %= MO;
    }
    while(m--) {
        x = readint(); y = readint();
        ans -= query(x, a[x], v[x]); ans = (ans % MO + MO) % MO;
        erase(x, a[x], v[x]);
        ans -= query(y, a[y], v[y]); ans = (ans % MO + MO) % MO;
        erase(y, a[y], v[y]);
        std::swap(a[x], a[y]);
        std::swap(v[x], v[y]);
        add(x, a[x], v[x]);
        ans += query(x, a[x], v[x]); ans %= MO;
        add(y, a[y], v[y]);
        ans += query(y, a[y], v[y]); ans %= MO;
        printf("%lld\n", ans);
    }
    return 0;
}