A. Lucky?
题意
给定一个长度为 6 6 6 的字符串,判断前三位的加和与后三位的加和是否相等。
#include <bits/stdc++.h>
#define int long long
using namespace std;
void solve(){
string s;
cin>>s;
int a=(s[0]-'0')+(s[1]-'0')+(s[2]-'0');
int b=(s[3]-'0')+(s[4]-'0')+(s[5]-'0');
if(a==b)
cout<<"YES\n";
else
cout<<"NO\n";
}
signed main(){
int t;
cin>>t;
while(t--)
solve();
return 0;
}
B - Equal Candies
题意
n n n 堆糖果,可以从每堆糖果中取走任意个,问总共最少需要取几个使得所有堆的糖果数量相等。
每堆最后只会剩下 m i n min min 个,因此取走 s u m − m i n ∗ n sum-min*n sum−min∗n 即可
#include <bits/stdc++.h>
#define int long long
using namespace std;
void solve(){
int n,x;
cin>>n;
int sum=0,minn=1e9;
for(int i=1;i<=n;i++){
cin>>x;
minn=min(x,minn);
sum+=x;
}
cout<<sum-minn*n<<endl;
}
signed main(){
int t;
cin>>t;
while(t--)
solve();
return 0;
}
C - Most Similar Words
题意
对于一个字符串,每次操作可以选中该串中的某个字符并将其更改为字典序中相邻的字符,现给定 n n n 个字符串,问使其中某两个字符串相等所需要的最小操作数是多少。
由于数据范围很小,考虑直接两两间暴力对比。
#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int N=105;
string x[N];
int s[N][N];
void solve(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>x[i];
for(int j=1;j<=m;j++)
s[i][j]=(int)(x[i][j-1]-'a'+1);
}
int minn=inf,now,tmp;
for(int a=1;a<=n;a++){
for(int b=a+1;b<=n;b++){
now=inf;
for(int k=1;k<=26;k++){
tmp=0;
for(int j=1;j<=m;j++)
tmp+=abs(s[a][j]-s[b][j]);
now=min(tmp,now);
}
minn=min(now,minn);
}
}
cout<<minn<<endl;
}
signed main(){
int t;
cin>>t;
while(t--)
solve();
return 0;
}
D - X-Sum
题意
给定一个 n ∗ m n*m n∗m 的矩阵,每个方格内都有一个非负整数,问该矩阵中选出一对对角线能够得到的最大值。
预处理两个方向上每条对角线得到的贡献值,则一对对角线得到的答案为两条不同方向的对角线之和,再减去重叠方格的值。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=500;
int g[N][N],l[N],r[N];
void init(){
memset(g,0,sizeof(g));
memset(l,0,sizeof(l));
memset(r,0,sizeof(r));
}
void solve(){
int n,m;
init();
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>g[i][j];
for(int k=1;k<n+m;k++)
for(int i=k,j=1;i>=1&&j<=m;i--,j++)
l[k]+=g[i][j];
for(int k=1;k<n+m;k++)
for(int i=k,j=m;i>=1&&j>=1;i--,j--)
r[k]+=g[i][j];
int maxx=0,now;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
now=(l[i+j-1]+r[i+m-j]-g[i][j]);
maxx=max(maxx,now);
}
}
cout<<maxx<<endl;
}
signed main(){
int t=1;
cin>>t;
while(t--)
solve();
return 0;
}
E - Eating Queries
题意
给定 n n n 个糖果,第 i i i 个糖果的甜度为 a i a_i ai ,问最少需要在这些糖果中选择几个,使得它们的甜度之和大于等于询问值。
由大到小排序后前缀和贪心处理,考虑到数据范围,采用二分找到符合的答案。
#include <bits/stdc++.h>
using namespace std;
const int N=2e5;
int a[N],s[N];
bool cmp(int a,int b){
return a>b;
}
int deal(int l,int r,int q){
if(s[r]<q)
return -1;
int mid=l+r>>1;
while(l<r){
if(s[mid]<q)
l=mid+1;
else
r=mid;
mid=l+r>>1;
}
return mid;
}
void solve(){
int n,q;
cin>>n>>q;
for(int i=1;i<=n;i++)
cin>>a[i];
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++)
s[i]=s[i-1]+a[i];
int x;
for(int i=1;i<=q;i++){
cin>>x;
cout<<deal(1,n,x)<<endl;
}
}
signed main(){
int t=1;
cin>>t;
while(t--)
solve();
return 0;
}
F - Longest Strike
题意
求一个最长的区间 [ l , r ] [l,r] [l,r] ,使得该区间内的所有数在数组 a a a 内出现的次数都不少于 k k k 次。
采用两个 m a p map map 进行标记, m p 1 mp1 mp1 记录每个数字出现的次数, m p 2 mp2 mp2 记录出现次数超过 k k k 次的数字,遍历得到 m p 2 mp2 mp2 中的最长连续区间即可。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
void solve(){
int n,k,x;
cin>>n>>k;
map<int,int> mp1;
map<int,int> mp2;
for(int i=0;i<n;i++){
cin>>x;
mp1[x]++;
}
for(auto i:mp1)
if(i.second>=k)
mp2[i.first]=i.second;
int l=-1,r=-1;
int ans=-1,ansl=-1,ansr=-1;
for(auto i:mp2){
if(l==-1&&r==-1){
l=i.first;
r=i.first;
}
else{
if(i.first!=r+1){
if(r-l>ans){
ans=r-l;
ansl=l;
ansr=r;
}
l=i.first;
r=i.first;
}
else
r++;
}
}
if(r-l>ans){
ans=r-l;
ansl=l;
ansr=r;
}
if(ansl==-1&&ansr==-1)
cout<<-1<<endl;
else
cout<<ansl<<" "<<ansr<<endl;
}
signed main(){
int t=1;
cin>>t;
while(t--){
solve();
}
return 0;
}
G. White-Black Balanced Subtrees
题意
一棵树上的每个结点只能是黑色或白色,问该树上有多少棵子树满足其中的黑色结点与白色结点的数量相同。
树形 d p dp dp , d f s dfs dfs 预处理以每个结点为根节点时,该子树内的黑色结点数量和白色结点数量。
#include<bits/stdc++.h>
using namespace std;
const int N=5000;
vector<int> h[N];
bool vis[N];
int white[N],black[N];
void dfs(int u){
vis[u]=1;
for(auto v:h[u]){
if(!vis[v]){
dfs(v);
white[u]+=white[v];
black[u]+=black[v];
}
}
}
void init(){
for(int i=0;i<=4000;i++){
h[i].clear();
vis[i]=0;
white[i]=black[i]=0;
}
}
void solve(){
init();
int n,ans=0,x;
cin>>n;
for(int i=2;i<=n;i++){
cin>>x;
h[x].push_back(i);
}
string s;
cin>>s;
for(int i=0;i<n;i++){
if(s[i]=='B')
black[i+1]=1;
else
white[i+1]=1;
}
dfs(1);
for(int i=1;i<=n;i++)
if(white[i]==black[i])
ans++;
cout<<ans<<endl;
}
signed main(){
int t=1;
cin>>t;
while(t--)
solve();
}
H2. Maximum Crossings (Hard Version)
题意
求 n n n 条线段间产生交叉点的数量。
通过树状数组求逆序对的数量,注意值相同也被视为产生贡献。
#include<bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;
const int N=2e5+10;
int n,m,k;
int tr[N];
int lowbit(int x){
return x&(-x);
}
int add(int x,int y){
for(int i=x;i<N;i+=lowbit(i))
tr[i]+=y;
}
int sum(int x){
int res=0;
for(int i=x;i;i-=lowbit(i))
res+=tr[i];
return res;
}
void solve(){
int ans=0;
cin>>n;
memset(tr,0,sizeof tr);
for(int i=1;i<=n;i++){
int q;
cin>>q;
ans+=sum(N)-sum(q-1);
add(q,1);
}
cout<<ans<<endl;
}
signed main(){
int t=1;
cin>>t;
while(t--)
solve();
return 0;
}