ATCODER abc240部分题解

A

判断两数是否相邻，或两数分别为1，10。

#include<bits/stdc++.h>
using namespace std;
int main(){
	int a,b;cin>>a>>b;
	if(abs(a-b)==1||abs(a-b)==9)cout<<"Yes";
	else cout<<"No";
}

B

查重，数不同数字的个数，使用set高效去重。

#include<bits/stdc++.h>
using namespace std;
set<int>s;int n;
int main() {
    cin >> n;
    for (int i = 1, x; i <= n; cin >> x, i++, s.insert(x));
    cout << s.size();
}

C

n组数，每组两个，每次从第i组任选1个数，最后n组选出的和为x
dfs,时间复杂度O(2^100)，明显超时

超时_
void dfs(int dep,int dis){
	if(dep==n+1){
		if(dis+a[dep]==x||dis+b[dep]==x)Yes();
		return;
	}
	dfs(dep+1,dis+a[dep]);
	dfs(dep+1,dis+b[dep]);
}

考虑剪枝，记忆化

#include<bits/stdc++.h>
using namespace std;
int a[110], b[110];
bool vis[110][10010];
int N, X;
void dfs(int n, int x) {
    if (n == N + 1 && x == X){cout<<"Yes";exit(0);}
    if (n == N + 1||vis[n][x]||x > X)return;//剪枝
    dfs(n + 1, x + a[n]); vis[n + 1][x + a[n]] = 1;
    dfs(n + 1, x + b[n]); vis[n + 1][x + b[n]] = 1;
}
int main(){
    cin >> N >> X;
    for(int i = 1; i <= N; i++)cin >> a[i] >> b[i];
    dfs(1, 0);
	cout<<"No";
}

ac了，甚至比dp快。我觉得这个更好理解。

考虑dp，设dp[i,j]前i次，可以到j距离。
当i=n,j=x时即为所求

#include<bits/stdc++.h>
using namespace std;
int n, x, a, b, dp[102][100002];
int main() {
	dp[0][0]=1;//设置出生点
    cin >> n >> x;
    for (int i = 0; i < n; i++) {
        cin >> a >> b;
        for (int j = 0; j <= x; j++) //dp[i,j]跳i次，能否到达j？1：0
            if (dp[i][j]) {
                if (j + a <= x)dp[i + 1][j + a] = 1;
                if (j + b <= x)dp[i + 1][j + b] = 1;
            }
    }
	if(dp[n][x])cout<<"Yes";
	else cout<<"No";
}

D

相同的数字n叠在一起n次会消掉，每次输出剩余个数。
很明显的栈，发现用stl很慢好像还会爆，果断手写，轻松通过。

这个题很像宝开游戏“祖玛”，相同的叠在一起一定个数会消掉。
（可恶，我怎么会知道的，一看题就想到了）

#include<bits/stdc++.h>
using namespace std;
int n, noww;
pair<int, int>p[(int)2e5 + 3];//记录数字和该数字连续出现的次数
int main() {
    cin >> n;
    for (int i = 1, num; i <= n; i++) {
        cin >> num;
        if (i == 1 || num != p[noww].first)p[++noww] = {num, 1};//不一样或者第一次就直接往里放
		else {
			noww++;
            p[noww] = {num, p[noww - 1].second + 1};//连续个数增加了1
            if (p[noww].first == p[noww].second)noww -= num;//应该消的时候直接退回n个之前的地址
        }
        cout << noww << endl;
    }
}

E

施工中