WA13求助
查看原帖
WA13求助
396033
zhengzixuan楼主2024/4/7 13:41

前缀和,但是只用一个变量存,不知道问题在哪

#include <iostream>
#include <iomanip>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <string>
#include <ctime>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#define int long long
const int N = 2e5 + 5;
const int INF = 0x3f3f3f3f;
using namespace std;
int n, q, x, now, sum, s[N], vis[N], ans[N];
signed main(){
    cin >> n >> q;
    memset(s, INF, sizeof s);
    while(q--){
        cin >> x;
        if (vis[x]){
            ans[x] += sum - s[x];
            s[x] = INF;
            now--;
            vis[x] = 0;
        } else{
            s[x] = sum;
            now++;
            vis[x] = 1;
        }
        sum += now;
    }
    for (int i = 1; i <= n; i++){
        if (s[i] != INF)
            ans[i] += sum - s[i];
        cout << ans[i] << ' ';
    }
    return 0;
}
2024/4/7 13:41
加载中...