Lang:G++
Edit12345678910111213141516171819202122232425262728293031#include <iostream>#include <stdio.h>#include <stdlib.h>#include <algorithm>#include <string.h>using namespace std;int a[200001];int cnt[17];int ans[6];int main(){int T,cas=1;int n,k;scanf("%d",&T);while(T--){scanf("%d %d",&n,&k);for(int i=0;i<n;i++)scanf("%d",&a[i]);memset(cnt,0,sizeof(cnt));memset(ans,0,sizeof(ans));for(int i=0;i<n;i++){if(a[i]==1){ans[0] += cnt[2]+cnt[3];ans[1] += cnt[2]+cnt[3];ans[2] += cnt[3];ans[4] += cnt[2];