AtCoder Beginner Contest 369 A-D 题解（9.5号以后出E）

先做的B，题目看错了，把数组直接排序，就慢了。做A没有考虑到 $A=B$ 的情况，又慢了。到C又了十想几分钟，叒满了。D是dp我一直在想贪心，叕慢了，到70分钟才做出4题，E可能是WarshallFloyd没学过，还在想最小生成树的时候比赛结束了……

好了不扯了回归正题。

A

题意

给你两个数 $A$ 和 $B$，再加一个 整数，使得它与$A,B$ 构成等差数列，问有几个数满足条件

思路

若 $A=B$，则答案为 $1$

否则，若 $\frac{A+B}{2}$ （中点公式）是整数，答案为 $3$，其余为 $2$

C++ 代码

#include<bits/stdc++.h>
using namespace std;
int a,b;
int main() {
    cin>>a>>b;
    if(a==b){
        cout<<1;
    }else{
        if((a+b)%2==0){
            cout<<3;
        }else{
            cout<<2;
        }
    }
}

B

题意

有一架钢琴，一只手从 $x$ 移动到 $y$ 疲劳度增加 $|x-y|$，一共 按顺序 弹奏 $n$ 个键（分左手和右手），问最终疲劳度是多少？

Notes：第一次按键不增加疲劳度，因为刚开始可以放在这个键上

思路

很明显，直接模拟，记得把左右手分开，最终答案每次加上这次与上次的绝对值

C++ 代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n;
vector<int> l,r;
signed main(){
	cin>>n;
	for(int i=0;i<n;i++){
		int ai;
		char c;
		cin>>ai>>c;
		if(c=='L'){
			l.push_back(ai);
		}else{
			r.push_back(ai);
		}
	}
	int ans=0;
	for(int i=1;i<l.size();i++){
		ans+=abs(l[i]-l[i-1]);
	}
	for(int i=1;i<r.size();i++){
		ans+=abs(r[i]-r[i-1]);
	}
	cout<<ans<<endl;
	return 0;
}

C

题意

给你 $n$ 个数，选择 $l,r(1\le l\le r\le n)$ ，使得 $a_l,a_{l+1},a_{l+2},...,a_r$ 是等差数列，问：有几个这样的 $(l,r)$

思路

找出每两项之间的公差，看每一种公差连续几个（设为 $len$ 个），则这一段的答案为 $C_{len}^2$，即 $\frac{len(len-1)}{2}$，最终把答案加起来

C++ 代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=200005;
int n;
int v[maxn];
int dif[maxn];
int ans;
signed main(){
	cin>>n;
	if(n==1){
		cout<<1<<endl;
		return 0;
	}
	for(int i=0;i<n;i++){
		cin>>v[i];
	}
	for(int i=1;i<n;i++){
		dif[i]=v[i]-v[i-1];
	}
	int len=2;
	vector<int> p;
	for(int i=2;i<n;i++){
		if(dif[i]!=dif[i-1]){
			p.push_back(len);
			len=2;
		}else{
			len++;
		}
	}
	p.push_back(len);
	for(int x:p){
		ans+=x*(x-1)/2;
	}
	cout<<ans+n<<endl;
	return 0;
}

D

题意

给你 $n$ 个数，你要对这些数按顺序选择，并遵循一下规则：

• 如果这个数是已选择的第偶数个，则 $ans+=2\cdot a_i$
• 否则 $ans+=a_i$
• 你也可以选择跳过这个数

思路

是动态规划不是贪心！！！！！

定义 $dp[n][2]$：

$$\{\begin{array}{l}dp[i][0]:到目前为止，第i个元素是所选的第奇数个时，ans的最大值\\ dp[i][1]:到目前为止，第i个元素是所选的第偶数个时，ans的最大值\end{array}$$
所以做如下转移：
$$\{\begin{array}{l}dp[1][0]=v[1]\\ dp[1][1]=0\\ dp[i][0]=\max (dp[i-1][1]+v[i],dp[i-2][1]+v[i])(2\le i\le n)\\ dp[i][1]=\max (dp[i-1][0]+v[i]\ast 2,dp[i-2][0]+v[i]\ast 2)(2\le i\le n)\end{array}$$
最终答案为 $\max (dp[n][1],dp[n][0])$

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=200005;
int n;
int v[maxn];
int dp[maxn][2];
int sum=0;
signed main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>v[i];
	}
	dp[1][0]=v[1];
	dp[1][1]=0;
	for(int i=2;i<=n;i++){
		dp[i][0]=max(dp[i-1][1]+v[i],dp[i-2][1]+v[i]);
		dp[i][1]=max(dp[i-1][0]+v[i]*2,dp[i-2][0]+v[i]*2);
	}
	cout<<max(dp[n][1],dp[n][0])<<endl;
	return 0;
}