Codeforces Round #789 (Div. 2)(A-D)

Codeforces Round #789 (Div. 2)
A. Tokitsukaze and All Zero Sequence

一个小小的思维贪心，很容易知道如果有0，直接用0去和其他数处理，如果没有就先创造一个

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int M = 1e9 + 7;
const int N = 2e6 + 9;
int a[N];
signed main()
{int t;cin>>t;while (t--){int n;cin>>n;int ff = 0;//有咩有0int cou = 0;int kk = 0;map<int,int>mp;for (int i=0;i<n;i++) {cin>>a[i];if (a[i]==0) ff = 1;else cou++;if (mp[a[i]]) kk = 1;mp[a[i]]++;}if (ff||kk) cout<<cou<<endl;else cout<<cou+1<<endl;}
}

B2. Tokitsukaze and Good 01-String (hard version)

B2覆盖了B1，所以直接发B2好了
这题当时没写出来，因为我最后写的这题，可能脑子不够用了，标签是dp，但是应该是可以贪心的。
首先，因为是偶数子串，所以a[i]==a[i+1]对于所有的奇数位应该成立，字符串可以分割成偶数个，两个一组，所以实际问题就转化成了去找1100或者0011问题，因为如果有0 10 1情况，中间变成谁都可以 1 10 01 01 10 0这种情况肯定是中间都变成一端最好

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int M = 1e9 + 7;
const int N = 2e6 + 9;
int a[N];
string s;
signed main()
{int t;cin >> t;while (t--){int ans = 0,res = 0,n;char ch = '3';cin>>n>>s;for (int i=0;i<n;i+=2) {if (s[i]!=s[i+1]) res++;else if (ch!=s[i]) {ch = s[i],ans++;}}cout<<res<<" "<<max(1ll,ans)<<endl;}
}

C. Tokitsukaze and Strange Inequality
这题算的上是一个树状数组的模板题，所以没啥好说的，会树状数组就会写
枚举b,c就行

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int M = 1e9 + 7;
const int N = 2e6 + 9;
int tr1[N],tr2[N],n,a[N];
void add1(int x)
{for (;x<=n;x+=(x&-x)) tr1[x]++;
}
void add2(int x)
{for (;x<=n;x+=(x&-x)) tr2[x]++;
}
int sum1(int x)
{int res = 0;for (;x;x-=(x&-x)) res+=tr1[x];return res;
}
int sum2(int x)
{int res = 0;for (;x;x-=(x&-x)) res+=tr2[x];return res;
}
void init()
{for (int i=0;i<=n;i++) tr1[i] = tr2[i] = 0;
}
signed main()
{int t;cin>>t;while (t--){cin>>n;init();int res = 0;for (int i=1;i<=n;i++) cin>>a[i];for (int i=1;i<=n;i++){for (int i=0;i<=n;i++) tr2[i] = 0;for (int j=n;j>i;j--){res+=sum1(a[j]-1)*sum2(a[i]-1);add2(a[j]);}add1(a[i]);}cout<<res<<endl;}
}

D. Tokitsukaze and Meeting
这题看起来复杂，因为进去一个人，所有人都要变，但可以转化一下，从两个维度看。
第一个就是对于行的贡献，每进去一个人去和上一行的这个位置的人进去的时候比较就行，如果中间确实有1，那这行就是有1的，如果没有就没有
列就更好算了，因为每个人都是对应的，1一定和m+1在同一列，所以这列没有1就加一个，如果有了就不管了。

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int M = 1e9 + 7;
const int N = 2e6 + 9;
int a[N],n, m;
string s;
signed main()
{int t;cin >> t;while (t--){cin >> n >> m >> s;for (int i=1;i<=m;i++) a[i] = 0;int f[n+1][m+1];memset(f, 0, sizeof f);//先处理了行的贡献int cou = 0;for (int i = 1; i <= n; i++)for (int j = 1; j <= m; j++){int w = (i-1) * m + j  - 1;if (s[w] == '1') cou = m;f[i][j] = f[i - 1][j] + (cou > 0);cou--;}//再处理了列的int cnt = 0;for (int i = 1; i <= n; i++)for (int j = 1; j <= m; j++){int w = (i-1)*m + j - 1;if (s[w]=='1'&&!a[j]) a[j] = 1,cnt++;cout<<cnt+f[i][j]<<" ";}cout<<endl;}
}