【套题】Codeforces#304(div2)

A. Soldier and Bananas

• 题意：第一个香蕉要$k$刀，第二个$2k$刀，第$i$个要$i*k$刀。现有$n$刀，问可以买几个香蕉。
• 题解：等差数列求和，我们知道只需要找到$p$使得 $$\sum_{i=1}^{p}i*k \le n \lt \sum_{i=1}^{p+1}{i*k}$$ 即可，移项就可以得到公式，同时上述求和公式是单调递增的，因此也可以二分答案。
• 参考代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
inline ll foo(int x, int k) {
    return (((1LL + x) * x) >> 1) * k;
}

ll gao(ll k, ll n, ll w) {
    return max(foo(w, k) - n, 0LL);
}

int main() {
    int k, n, w;
    while (~scanf(" %d %d %d", &k, &n, &w)) {
        printf("%I64d\n", gao(k, n, w));
    }
    return 0;
}

B. Soldier and Badges

• 题意：给$n$个数，$1\le a_i\le n$，将一个数从$a$变大成$b$需要消耗$b-a$个金币，问如何把这堆数变成两两不等的，顺序任意。
• 题解：直接扫一遍，对一个数$a$，向上找一个之前没有占用的位置填充即可。
• 参考代码：

#include <cstdio>
#include <cstring>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

const int MAX = 6007;
bool vis[MAX];
int foo(int x) {
    int res = 0;
    while (vis[x]) {
        ++x;
        ++res;
    }
    vis[x] = true;
    return res;
}

int main() {
    int n;
    while (~scanf(" %d", &n)) {
        memset(vis, false, sizeof(vis));
        int a, sum = 0;
        for (int i = 1; i <= n; ++i) {
            scanf(" %d", &a);
            sum += foo(a);
        }
        printf("%d\n", sum);
    }
    return 0;
}

C. Soldier and Cards

• 题意：两堆牌，每次取最上面的两张比较，如果更大，则吃掉另一张，并将其放到底部，然后将自己这张也放到底部。一张牌都没有时为输。问游戏能不能终止，如果能，输出谁胜利以及需要进行几轮。
• 题解：考虑到总牌数并不多($k_1+k_2=n$, $2\le n \le 10$)，状态数最多只有$A_{10}^{10}*C_9^1$这么多，实际上要少得多，因此直接状态压缩模拟即可。
• 参考代码：

#include <cstdio>
#include <cstring>
#include <set>
#include <map>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

struct State {
    vector<int> left;
    vector<int> right;
    bool operator==(const State& B)const {
        if (left.size() != B.left.size()
            || right.size() != B.right.size()) {
            return false;
        }
        for (int i = 0; i < left.size(); ++i) {
            if (left[i] != B.left[i]
                || right[i] != B.right[i]) {
                return false;
            }
        }
        return true;
    }
    bool operator<(const State& B)const {
        if (left.size() != B.left.size()) {
            return left.size() < B.left.size();
        }
        for (int i = 0; i < left.size(); ++i) {
            if (left[i] != B.left[i]) {
                return left[i] < B.left[i];
            }
        }
        if (right.size() != B.right.size()) {
            return right.size() < B.right.size();
        }
        for (int i = 0; i < right.size(); ++i) {
            if (right[i] != B.right[i]) {
                return right[i] < B.right[i];
            }
        }
        return false;
    }
    void print() {
        printf("{\n\t%d", left.size());
        for (int i = 0; i < left.size(); ++i) {
            printf(" %d", left[i]);
        }
        printf("\n\t%d", right.size());
        for (int i = 0; i < right.size(); ++i) {
            printf(" %d", right[i]);
        }
        puts("\n}\n");
    }
};

map<State, bool> M;
pair<int, int> gao(State& thus) {
    M.clear();
    for (int i = 0; ; ++i) {
        //puts("----------------------");
        //thus.print();
        //getchar();
        if (thus.left.empty()) {
            return make_pair(2, i);
        } else if (thus.right.empty()) {
            return make_pair(1, i);
        } else if (M.find(thus) != M.end()) {
            return make_pair(-1, i);
        } else {
            M.insert(make_pair(thus, true));
            //generate the next thus
            if (thus.left[0] > thus.right[0]) {
                thus.left.push_back(thus.right[0]);
                thus.left.push_back(thus.left[0]);
            } else {
                thus.right.push_back(thus.left[0]);
                thus.right.push_back(thus.right[0]);
            }
            thus.left.erase(thus.left.begin());
            thus.right.erase(thus.right.begin());
        }
    }
}

int main() {
    int n;
    State my;

    while (~scanf(" %d", &n)) {
        int m, a;
        scanf(" %d", &m);
        my.left.clear();
        while (m--) {
            scanf(" %d", &a);
            my.left.push_back(a);
        }
        scanf(" %d", &m);
        my.right.clear();
        while (m--) {
            scanf(" %d", &a);
            my.right.push_back(a);
        }
        pair<int, int> res = gao(my);
        if (res.first == -1) {
            puts("-1");
        } else {
            printf("%d %d\n", res.second, res.first);
        }
    }

    return 0;
}

D. Soldier and Number Game

• 题意：问${{a!} \over {b!}}$分解质因数后又多少个质因子。如$28=2^2*7$，有3个质因子，注意，这里质因子$2$的指数为$2$，要算成2个。
• 题解：记$\Omega(n)$为$n$的质因子个数，则$\Omega(1)=0$。根据分解质因数后的形式，设： $$n=x*y$$ 则显然有： $$\Omega(n)=\Omega(x)+\Omega(y)$$ $\Omega(n)$有这种性质，我们就称$\Omega(n)$是加性函数。同时可以知道： $$\Omega(x)=\Omega(n)-\Omega(y)$$ 于是得解：
  (1) 首先用筛法筛出质数，求出所有质数的$\Omega$函数值为1(记为$prime[x]=1$)，并且对任意一个合数$y$得出其一个质因子$p$，记为$flag[y]$，即$flag[y]=p$。
  (2) 然后对$1\le x\le 5000000$，我们线性地扫一遍： $$prime[x]=prime[x/flag[x]] + prime[flag[x]]$$ 其中$prime[x]=\Omega(x)$。
  (3) 由于 $${{a!} \over {b!}}={{\sum_{i=1}^{a}\Omega(i)}\over {\sum_{i=1}^{b}\Omega(i)}}$$ 分子和分母都是前缀和的形式，令： $$dp[x]=\sum_{j=1}^{x}\Omega(j)$$ 就可以得到答案： $$ans(a,b)=dp[a]-dp[b]$$
• 参考代码：

#include <bits/stdc++.h>
using namespace std;

const int MAX = 5000007;
int flag[MAX] = {0};
int prime[MAX] = {0}; //prime[i]: i的质因数的个数
int dp[MAX] = {0}; //前缀和

void init() {
    int i;
    for (i = 2; i <= MAX / i; ++i) {
        if (flag[i] == 0) {
            flag[i] = i;
            prime[i] = 1;
            //printf("flag[%d] = %d\n", i, flag[i]);
            //getchar();
            for (int j = i * i; j < MAX; j += i) {
                if (flag[j] == 0) {
                    flag[j] = i;
                    //printf("flag[%d] = %d\n", j, i);
                    //getchar();
                }
            }
        }
    }
    for (; i < MAX; ++i) {
        if (flag[i] == 0) {
            flag[i] = i;
            prime[i] = 1;
        }
    }

    for (i = 2; i < MAX; ++i) {
        prime[i] = prime[i / flag[i]] + prime[flag[i]];
    }

    for (i = 1; i < MAX; ++i) {
        dp[i] = dp[i - 1] + prime[i];
    }
    //for (int i = 1; i < 10; ++i) {
    //  printf("prime[%d] = %d, dp[%d] = %d\n", i, prime[i], i, dp[i]);
    //}
}

int main() {
    init();

    int T, a, b;
    scanf(" %d", &T);
    while (T--) {
        scanf(" %d %d", &a, &b);
        printf("%d\n", dp[a] - dp[b]);
    }
    return 0;
}

E. Soldier and Traveling

• 题意：有$n$个城市，$m$条道路（无向），城市里有一些士兵，问在每个士兵只能走一次的情况下，能不能把原来的各城市的士兵数$(a_1,a_2,...a_n)$变成$(b_1,b_2,...,b_n)$。如果能给出任一个调整方案。
• 题解：网络流。建图时，增加源点$source$，汇点$dest$，每个城市作为一个点，同时拆点为$i$何$i+n$，如果有道路$(x,y)$，那么建边$x\to n+y$ 和 $y\to n+x$，容量分别为$a_x$和$a_y$。同时，源点和点$i_{1\le i\le n}$之间建边，容量为$a_x$；汇点和点$i_{{n+1}\le i\le {2*n}}$之间建边，容量为$b_i$。得到$source$到$dest$的最大流$flow$后，判断$flow$是否等于所有点的士兵总和$\sum_{i=1}^{n}a_i$即可。当然，也应该要判断$\sum_{i=1}^{n}a_i$和$\sum_{i=1}^{n}b_i$是否相等。
• 参考代码：

#include <bits/stdc++.h>
using namespace std;

const int INF = 0xfffffff;
const int MAX = 107 << 1;
struct Edge {
    int to;
    int flow;
    int cap;
    int rev; //mapto Edge G[to][rev]
};
vector<Edge> G[MAX];
int level[MAX];
//store answer
int ans[MAX][MAX];
int n, m;
int a[MAX], b[MAX];

void addEdge(int from, int to, int cap) {
    G[from].push_back((struct Edge){to, 0, cap, G[to].size()});
    G[to].push_back((struct Edge){from, cap, cap, G[from].size() - 1});
}

bool bfs(int s, int t) {
    memset(level, -1, sizeof(level));
    level[s] = 0;
    queue<int> Q;
    Q.push(s);
    while (!Q.empty()) {
        int p = Q.front();
        Q.pop();

        for (vector<Edge>::iterator it = G[p].begin(); it != G[p].end(); ++it) {
            if (level[it->to] == -1 && it->flow < it->cap) {
                Q.push(it->to);
                level[it->to] = level[p] + 1;
                if (it->to == t) {
                    return true;
                }
            }
        }
    }
    //printf("No Road.\n");
    return false;
}

int dfs(int s, int t, int flow) {
    if (s == t) {
        return flow;
    }
    int sum = 0, tmp;
    for (vector<Edge>::iterator it = G[s].begin(); it != G[s].end(); ++it) {
        if (level[it->to] == level[s] + 1 && it->flow < it->cap) {
            tmp = dfs(it->to, t, min(flow, it->cap - it->flow));
            //printf("gao(%d, %d), fetch %d\n", s, it->to, tmp);
            it->flow += tmp;
            G[it->to][it->rev].flow -= tmp;
            sum += tmp;
            flow -= tmp;
        }
    }
    return sum;
}

int dinic(int s, int t) {
    int sum = 0;
    while (bfs(s, t)) {
        sum += dfs(s, t, INF);
    }
    return sum;
}

inline int ex(int x) {
    return (x >= 1 && x <= n) ? x : x - n;
}

int main() {
    while (~scanf(" %d %d", &n, &m)) {
        for (int i = 0; i < MAX; ++i) {
            G[i].clear();
        }

        int source = 0, dest = n << 1 | 1;
        int sum1 = 0, sum2 = 0;
        for (int i = 1; i <= n; ++i) {
            scanf(" %d", a + i);
            sum1 += a[i];
            addEdge(source, i, a[i]);
            addEdge(i, i + n, a[i]);
        }
        for (int i = 1; i <= n; ++i) {
            scanf(" %d", b + i);
            sum2 += b[i];
            addEdge(i + n, dest, b[i]);
        }

        int x, y;
        for (int i = 1; i <= m; ++i) {
            scanf(" %d %d", &x, &y);
            //match point <x, n + y>
            addEdge(x, n + y, a[x]);
            addEdge(y, n + x, a[y]);
        }

        int flow = dinic(source, dest);
        //printf("flow = %d\n", flow);

        if (sum1 != sum2 || flow != sum2) {
            puts("NO");
        } else {
            puts("YES");
            memset(ans, 0, sizeof(ans));
            for (int i = 1; i <= n; ++i) {
                ans[i][i] = a[i];
            }
            /*
            for (int i = source; i <= dest; ++i) {
                for (int j = 0; j < G[i].size(); ++j) {
                    printf("(%d, %d) -> %d\n", i, G[i][j].to, G[i][j].flow);
                }
            }
            puts("----");
            */
            for (int i = 1; i <= n; ++i) {
                for (int j = 0; j < G[i].size(); ++j) {
                    Edge& e = G[i][j];
                    if (e.flow > 0) {
                        ans[i][i] -= e.flow;
                        ans[i][e.to - n] += e.flow;
                    }
                }
            }
            for (int i = 1; i <= n; ++i) {
                for (int j = 1; j <= n; ++j) {
                    printf("%d%c", ans[i][j], j == n ? '\n' : ' ');
                }
            }
        }
    }

    return 0;
}