A. Dawid and Bags of Candies
乱搞。
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e5 + 5;
int a[4];
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
for(int i = 0; i < 4; i++) cin >> a[i];
sort(a, a + 4);
if(a[0] + a[3] == a[1] + a[2] || a[0] + a[1] + a[2] == a[3]) cout << "YES";
else cout << "NO";
return 0;
}
B. Ania and Minimizing
贪心。
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 2e5 + 5;
int n, k;
char s[N];
void run() {
cin >> s + 1;
if(n == 1 && k >= 1) {
cout << 0 << '
'; return;
}
for(int i = 1; i <= n; i++) {
if(i == 1) {
if(s[i] > '1' && k) {
s[i] = '1'; --k;
}
} else {
if(k && s[i] != '0') {
s[i] = '0'; --k;
}
}
}
cout << s + 1 << '
';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
while(cin >> n >> k) run();
return 0;
}
C. Anadi and Domino
题意：
存在一张图有(n,nleq 7)个点，现在要给他们染上(1)~(6)的颜色。最后问怎样染色，能够使得((c_1,c_2))这样的对数最多（不计算重复，不考虑顺序，(c_1,c_2)表示颜色）。
思路：
一开始想的是随机乱搞一发，结果出了不知道什么错wa了一个点= =
其实直接(dfs)就行了，复杂度不会超。
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 10;
int n, m;
int ans, res;
vector <int> g[N];
int c[N];
bool chk[N][N];
void dfs(int x) {
if(x == n + 1) {
res = 0;
for(int i = 1; i <= n; i++) {
for(auto it : g[i]) {
chk[c[i]][c[it]] = 1;
}
}
for(int i = 1; i <= 6; i++) {
for(int j = i; j <= 6; j++) {
if(chk[i][j]) {
++res;
chk[i][j] = 0;
}
}
}
ans = max(ans, res);
return;
}
for(int i = 1; i <= 6; i++) {
c[x] = i;
dfs(x + 1);
}
}
void run() {
for(int i = 1; i <= m; i++) {
int u, v; cin >> u >> v;
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1);
cout << ans << '
';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
while(cin >> n >> m) run();
return 0;
}
D. Marcin and Training Camp
题意：
先有(n)个人，每个人有两个属性：(a,b)。其中(a)的含义是，当前这个人会一些技能，技能的(id)就为(a)里面二进制(1)的位置。
(i)看不起(j)，当且仅当(i)会一项(j)不会的技能。
现在要选择一些人出来作为一支团队，要求不存在一个人看不起其它所有人并且(b)的和最大。
思路：
题意不好描述啊感觉= =
首先可以发现一个比较显然的结论：
- 团队中最大的(a)值至少有两个。
因为要(b)和最大，所以有推论：
- 若存在多个人的(a)相同，那么他们都可以选择。
处理完上面的情况，剩下的就是(a)只出现一次了。接下来又有一个结论：
- 一个(a)能够被选择，当且仅当他作为已选集合的一个子集。
如果不满足上面的条件，那么他就能够看不起其它所有人，除非有其他人(a)值和他相同，但这种情况已经排除了。
然后随便写写就好了。
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 7006;
struct People{
ll a, b;
bool operator < (const People &A) const {
if(a == A.a) return b < A.b;
return a < A.a;
}
}p[N], t[N];
int n;
void run() {
for(int i = 1; i <= n; i++) cin >> t[i].a;
for(int i = 1; i <= n; i++) cin >> t[i].b;
sort(t + 1, t + n + 1);
ll ans = 0;
int tot = 0;
vector <ll> S;
for(int i = 1, j; i <= n; i = j) {
ll tmp = 0, cnt = 0;
j = i;
while(j <= n && t[i].a == t[j].a) {
tmp += t[j].b;
++cnt; ++j;
}
if(cnt > 1) {
ans += tmp;
S.push_back(t[i].a);
} else {
p[++tot] = t[i];
}
}
for(int i = 1; i <= tot; i++) {
int ok = 0;
for(auto it : S) {
if((p[i].a & it) == p[i].a) ok = 1;
}
if(ok) {
ans += p[i].b;
// S.push_back(p[i].a);
}
}
cout << ans << '
';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
while(cin >> n) run();
return 0;
}
E. Kamil and Making a Stream
题意：
给出一颗树，每个结点都有其权值，先求所有链的(gcd)的和。
思路：
- 比较直接的想法就是(O(n^2))的暴力，对于每个点(v)，找到其所有的祖先(u)，计算(gcd)。
- 稍微改进一下，不用找到所有的祖先，只需要找到其父亲与所有祖先的(gcd)值就行，但复杂度还是没有变。
- 发现(gcd)的个数很少，大概为(log_{2}a[v])个，因为每个(gcd)都为(a[v])因子。所以直接用一个(map)来存一下出现次数就行了。
似乎这里(map)的(log)和(gcd)的(log)的分开的...
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e5 + 5, MOD = 1e9 + 7;
int n;
ll a[N];
vector <int> g[N];
unordered_map <ll, int> mp[N];
ll ans;
void dfs(int u, int fa) {
for(auto it : mp[fa]) {
ll now = __gcd(it.fi, a[u]);
mp[u][now] += it.se;
}
++mp[u][a[u]];
for(auto it : mp[u]) {
ans = (ans + it.fi * it.se % MOD) % MOD;
}
for(auto v : g[u]) {
if(v != fa) dfs(v, u);
}
}
void run() {
for(int i = 1; i <= n; i++) cin >> a[i];
for(int i = 1; i < n; i++) {
int u, v; cin >> u >> v;
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1, 0);
cout << ans << '
';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
while(cin >> n) run();
return 0;
}
F. Konrad and Company Evaluation
题意：
给出一个(n)个点(m)条边的有向图，现在有(q)次操作，每次可以选择一个点，然后将它所有的入边反转。
问每次操作过后形成的“三元环”数量，这里“三元环”的定义为对于点(a,b,c)，有(a->b->c)，那么这就算作一个。
(n,m,qleq 10^5)。
思路：
- 统计三元环时，易知一个点的贡献为(out[u]*in[u])，相当于枚举中点。
问题在于每次如何翻转边并且快速统计答案。然后题解就是暴力翻转就行了...
下面简略证明一下：
- 我们按照每个点的度从大到小排序，并且以度数为(sqrt{2m})划分界限，左边为“大点”，右边为“小点”。
- 易知左边点的个数不会超过(sqrt{2m})个，而右边点的度数不会超过(sqrt{2m})。
- 现在将所有边分为三类：在“大点”间的，在“小点”间的，以及跨过中线的，然后依次分析：
- 显然每次对“小点”操作不会超过(sqrt{2m})；
- 对于跨过中间的，我们先消耗一定复杂度让其全部指向“小点”，易知复杂度不超过(O(m))。
- 那么每次对“大点”操作，第一类复杂度不会超过(sqrt{2m})（因为个数只有那么多），对于第三类边，只有可能一开始已经翻转过，那么这类边每次操作的复杂度最高为(O(T))，(T)表示当前时间。所以对于一个“大点”而言，操作一次的最高复杂度为(O(q+sqrt{2m}))，最多操作(sqrt{2m})次。
* 综上，总的复杂度为$O(qsqrt{2m}+m)$的样子。
然后暴力就行啦，感觉证明的思想核心还是对度数进行分块然后来分析，挺巧妙的~其实简单点想就是，对于每个点只有可能第一次翻转复杂度较高，后面翻转就跟时间有关了，因为每次翻转对于一个点入度最多增加(1)。
代码很简单：
Code
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
//#define Local
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e5 + 5;
int n, m;
vector <int> in[N];
int d[N];
void run() {
for(int i = 1; i <= m; i++) {
int u, v; cin >> u >> v;
if(u > v) swap(u, v);
++d[u], ++d[v];
in[u].push_back(v);
}
ll ans = 0;
for(int i = 1; i <= n; i++) ans += 1ll * (d[i] - sz(in[i])) * sz(in[i]);
int q; cin >> q;
while(q--) {
cout << ans << '
';
int x; cin >> x;
ans -= 1ll * (d[x] - sz(in[x])) * sz(in[x]);
for(auto v : in[x]) {
ans -= 1ll * (d[v] - sz(in[v])) * sz(in[v]);
in[v].push_back(x);
ans += 1ll * (d[v] - sz(in[v])) * sz(in[v]);
}
in[x].clear();
}
cout << ans << '
';
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
#ifdef Local
freopen("../input.in", "r", stdin);
freopen("../output.out", "w", stdout);
#endif
while(cin >> n >> m) run();
return 0;
}